keerthi-balaji commited on
Commit
c1ee0d0
·
verified ·
1 Parent(s): 6aab689

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ generator = pipeline("text-generation", model="gpt2")
5
 
6
  def generate_prompt(topic):
7
  prompt = f"Write a creative story about {topic}. Here is a prompt to get you started:"
8
- result = generator(prompt, max_length=80, num_return_sequences=1)
9
  return result[0]['generated_text']
10
 
11
  iface = gr.Interface(
 
5
 
6
  def generate_prompt(topic):
7
  prompt = f"Write a creative story about {topic}. Here is a prompt to get you started:"
8
+ result = generator(prompt, max_length=100, num_return_sequences=1)
9
  return result[0]['generated_text']
10
 
11
  iface = gr.Interface(