Update app.py

app.py

@@ -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, truncation=True)[0]['generated_text']
+    story = story_generator(prompt, max_length=250, do_sample=True, temperature=0.8, top_p=0.9, repetition_penalty=1.2)[0]['generated_text']
     return story
 
 # 故事 → 語音（TTS）
@@ -50,7 +50,7 @@ if uploaded_file:
 
     # 讀取並顯示圖片
     image = Image.open(image_path)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
 
     # 生成圖片描述
     st.text("🔍 Generating image caption...")