Update app.py

app.py

@@ -27,8 +27,12 @@ def generate_caption(image):
 
 def generate_story(caption):
     # Generate the story based on the caption using the GPT-2 model
-    prompt = f"As a storyteller, please based on the image described as '{caption}', create a interesting and attractive story for children aged 3-10. Keep it positive and happy in tone. Also, add more imagination, and starting with wordings'one upon a time'"
-    story = story_generator(prompt, max_length=100, num_return_sequences=1)[0]["generated_text"]
+    prompt = f"Once upon a time, based on the image described as '{caption}', here is a short and interesting story for children aged 3-10. The story is positive and happy in tone, with added imagination:\n\n"
+    story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
+    
+    # Extract the story text from the generated output
+    story = story.split("\n\n")[1].strip()
+    
     return story
 
 def convert_to_audio(story):