Update app.py

app.py

@@ -29,7 +29,7 @@ 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 inspired by the image of {caption}. Here's the story:\n\n"
+    prompt = f"{caption}\n\n"
     story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
     
     # Extract the story text from the generated output