Update main.py

main.py

@@ -198,10 +198,12 @@ def handle_image_message(event):
         line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
         client = ImgurClient(client_id, client_secret, access_token, refresh_token)
 
-        # 使用翻譯任務
-        translator = pipeline("translation", model='Helsinki-NLP/opus-mt-zh-en') 
-        prompt = prompt[0]['translation_text']
-        image_binary = generate_image_hf(prompt)
+        translator = pipeline("translation", model='Helsinki-NLP/opus-mt-zh-en')
+        translation = translator(prompt_zh)
+        prompt_en = translation[0]['translation_text']
+
+        # 生成圖片
+        image_binary = generate_image_hf(prompt_en)
 
         if image_binary:
             album = "nvsYwgq"