Update app.py

app.py

@@ -59,10 +59,12 @@ def generate_story(caption, model, tokenizer):
     Returns:
         story: Generated story as a string (100 words or less).
     """
-    # Add a child-friendly story prompt
+    # Add a child-friendly story prompt with clear instructions
     story_prompt = (
         f"Once upon a time, in a magical land, {caption}. "
         "One sunny day, something magical happened! "
+        "The children were playing when they discovered something amazing. "
+        "Here is what happened next: "
     )
     input_ids = tokenizer.encode(story_prompt, return_tensors="pt")
     output = model.generate(