Update app.py

app.py

@@ -8,8 +8,8 @@ def img2text(url):
 
 def text2story(text):
     story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator", device_map="auto", return_full_text=False)
-    prompt = f"A short story under 100 words about: {text}."
-    generated = story_generator(text, max_length=100, max_new_tokens=120, do_sample=True, temperature=0.7, num_return_sequences=1)
+    prompt = f"A short story under 100 English words about: {text}."
+    generated = story_generator(text, max_new_tokens=130, do_sample=True, temperature=0.7)
     story_text = generated[0]['generated_text']
     return story_text