Update app.py

app.py

@@ -16,9 +16,8 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    prompt = f"Write a children's story for ages 3-10 based on: {text}"
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2", max_new_tokens=160, min_new_tokens=130, num_return_sequences=1)
-    story_text = pipe(prompt, return_full_text=False)[0]['generated_text']
+    story_text = pipe(text)[0]['generated_text']
     return story_text