Update app.py

app.py

@@ -27,8 +27,8 @@ def generate_caption(image):
 
 def generate_story(caption):
     # Generate the story based on the caption using the GPT-2 model
-    prompt = f"Write a short, simple children's story in 100 words based on the following image description:\n\n{caption}\n\nStory:"
-    story = story_generator(prompt, max_length=200, num_return_sequences=1)[0]["generated_text"]
+    prompt = f"Write a short, simple children's story approximately 100 words based on the following image description:\n\n{caption}\n\nStory:"
+    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("Story:")[1].strip()