Update app.py

app.py

@@ -58,8 +58,8 @@ def story_from_caption(caption: str, pipe) -> str:
     if not caption:
         return "Could not generate a story without a caption."
     prompt = (
-        f"Write an imaginative story 50–100 word story based only on {caption}"
-        f"Do not add new details."
+        f"Write an imaginative story 50–100 word story based on the caption: {caption}"
+        f"Be creative within the caption."
     )
     results = pipe(
         prompt,
@@ -168,4 +168,4 @@ with st.spinner("Synthesizing audio..."):
         st.warning("Audio generation failed.")
     else:
         st.audio(audio_bytes, format="audio/wav")
-st.balloons()
+st.balloons()