Update app.py

app.py

@@ -17,7 +17,7 @@ def load_caption_model():
 
 @st.cache_resource
 def load_story_model():
-    model_name = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
+    model_name = "Qwen/Qwen2.5-1.5B-Instruct"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
     return pipeline("text-generation", model=model, tokenizer=tokenizer)