Update app.py

app.py

@@ -25,7 +25,8 @@ def generate_caption(image):
     return caption
 
 def generate_story(caption):
-    prompt = f"Based on the image described as '{caption}', here is a short, simple, and engaging story for children aged 3-10. The story should be easy to understand, use age-appropriate language, and convey a positive message. Focus on the main elements in the image and create a story that sparks their imagination:\n\n"
+    # Generate the story based on the caption using the GPT-2 model
+    prompt = f"Once upon a time, in a world inspired by the image described as '{caption}', a delightful story unfolded for children aged 3-10. The story should be engaging, easy to understand, and use age-appropriate language. Let's begin the story:\n\n"
     story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
     
     # Extract the story text from the generated output