Update app.py

app.py

@@ -38,7 +38,11 @@ def compose_story_from_caption(caption_detail):
     )
     story_results = story_pipeline(prompt_text, num_return_sequences=1)
     story_text = story_results[0]['generated_text']
-    return story_text
+    if "Story:" in story_text:
+         story = story_text.split("Story:", 1)[1].strip()
+     else:
+         story = story_text.strip()
+    return story
 
 
 def convert_text_to_audio(text_content, audio_path="output.mp3"):