Update app.py

app.py

@@ -4,13 +4,13 @@ import string
 from transformers import pipeline
 
 def img2text(url):
-    image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large", use_fast=True)
+    image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base", use_fast=True)
     text = image_to_text_model(url)[0]["generated_text"]
     return text
 
 def text2story(text):
-    story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B", device_map="auto", return_full_text=False)
-    prompt = f"Write a short story under 100 words about the scenario: {text}."
+    story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-1.5B", device_map="auto", return_full_text=False)
+    prompt = f"Write a short story under 100 words based upon: {text}."
     generated = story_generator(prompt, max_new_tokens=150, do_sample=True)
     story_text = generated[0]['generated_text']