Update app.py

app.py

@@ -5,7 +5,7 @@ from gtts import gTTS
 
 st.set_page_config(page_title="Image to Audio Story", page_icon="🦜")
 caption_pipeline = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-story_pipeline = pipeline("text-generation", model="Qwen/Qwen2-1.5B")
+story_pipeline = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B-Instruct")
 
 
 def extract_image_caption(image_data):
@@ -19,7 +19,7 @@ def compose_story_from_caption(caption_detail):
         "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. "
         "Let your creativity shine by introducing engaging characters, adventurous journeys, and delightful surprises. "
-        "Your story should be vivid, original, and between 100 and 300 words in length.\n\n"
+        "Your story should be vivid, original, and less than 100 words in length.\n\n"
         f"Image Details: {caption_detail}\n\nStory:"
     )
     story_results = story_pipeline(prompt_text, num_return_sequences=1)