Update main.py

main.py

@@ -17,7 +17,7 @@ from imgurpython import ImgurClient
 import requests
 import tempfile
 from datetime import datetime
-from transformers import pipeline,MarianMTModel, MarianTokenizer
+from transformers import pipeline, MarianMTModel, MarianTokenizer
 
 
 # 設定 Google AI API 金鑰
@@ -129,14 +129,16 @@ def generate_image_hf(prompt):
 
 def translate_zh_to_en(text):
     model_name = "Helsinki-NLP/opus-mt-zh-en"
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
-    model = MarianMTModel.from_pretrained(model_name)
+
+    tokenizer = MarianTokenizer.from_pretrained(model_name, force_download=True)
+    model = MarianMTModel.from_pretrained(model_name, force_download=True)
     
     inputs = tokenizer(text, return_tensors="pt", padding=True)
     translated = model.generate(**inputs)
     result = tokenizer.decode(translated[0], skip_special_tokens=True)
     
     return result
+    
 #==========================
 # 使用者上傳圖片
 #==========================