Update app.py

app.py

@@ -12,6 +12,10 @@ def img2text(url):
 
 # text2story
 def text2story(text):
+    prompt = (
+        "Generate a fun, engaging, and kid-friendly story for children aged 3-10. "
+        "Based on the following scenario, create a simple, imaginative narrative: " + text
+    )
     story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
     story_text = story_generator(text, max_length=100, num_return_sequences=1)
     return story_text[0]["generated_text"]