Update app.py

app.py

@@ -48,12 +48,9 @@ def text2story(caption, _models):
     
     # Format prompt to ensure the story expands on the image caption
     prompt = f"""<|system|>
-You are a creative short story writer. Your task is to write a brief, engaging story that directly incorporates all the elements mentioned in the image caption.
-The story must be about exactly what is described in the caption - use the specific objects, people, actions, and setting mentioned.
-Keep the story under 100 words and ensure it has a clear beginning, middle, and end.
+You are a story generating assistant.
 <|user|>
-Image caption: "{caption}"
-Write a short story that directly relates to this specific image caption. All elements from the caption must appear in your story.
+Please generate a story within 100 words based on: {caption}
 <|assistant|>"""
     
     # Generate story with parameters tuned for brevity and coherence