Update app.py

app.py

@@ -7,7 +7,7 @@ os.system("pip install transformers==4.36.2")
 
 # 加載 Hugging Face 的模型
 image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-story_generator = pipeline("text-generation", model="facebook/opt-1.3b")
+story_generator = pipeline("text-generation", model="facebook/opt-350m", device=-1)
 
 # 圖片 → 文字（生成描述）
 def img2text(image_path):
@@ -17,7 +17,7 @@ def img2text(image_path):
 # 文字 → 故事（生成完整故事）
 def text2story(text):
     prompt = f"Write a fun and magical children's story based on this idea: {text}.\n\nOnce upon a time..."
-    story = story_generator(prompt, max_length=250, do_sample=True, temperature=0.8, top_p=0.9, repetition_penalty=1.2)[0]['generated_text']
+    story = story_generator(prompt, max_length=150, do_sample=True, temperature=0.8, top_p=0.9, repetition_penalty=1.2)[0]['generated_text']
     return story
 
 # 故事 → 語音（TTS）