Update app.py

app.py

@@ -15,13 +15,6 @@ def extract_image_caption(image_data):
 
 
 def compose_story_from_caption(caption_detail):
-    """
-    根据图像描述创作一篇充满创意的儿童故事。
-    """
-    story_pipeline = pipeline(
-        "text-generation",
-        model="Qwen/Qwen2-1.5B",
-    )
     prompt_text = (
         "You are a talented and imaginative storyteller for children aged 3 to 10. "
         "Using the details derived from the image below, craft a captivating tale that goes beyond merely describing the scene. "
@@ -33,12 +26,8 @@ def compose_story_from_caption(caption_detail):
     story_text = story_results[0]['generated_text']
     
     if "Story:" in story_text:
-        story = story_text.split("Story:", 1)[1].strip()
-    else:
-        story = story_text.strip()
-    
-    return story
-
+        return story_text.split("Story:", 1)[1].strip()
+    return story_text.strip()
 
 def convert_text_to_audio(text_content, audio_path="output.mp3"):
     """