Update main.py

main.py

@@ -21,9 +21,6 @@ import requests
 import tempfile
 from datetime import datetime
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-from transformers import pipeline
-
 
 # 設定 Google AI API 金鑰
 genai_gen.configure(api_key=os.environ["GOOGLE_API_KEY"])
@@ -115,18 +112,18 @@ def generate_image_with_gemini(prompt):
             model="gemini-2.0-flash-exp-image-generation",
             contents=prompt,
             config=types.GenerateContentConfig(
-            response_modalities=['Text', 'Image']
+            response_modalities=['Text', 'Image'])
             )
-        )
 
     for part in response.candidates[0].content.parts:
         if part.text is not None:
             print(part.text)
         elif part.inline_data is not None:
-          #image = PIL.Image.open(io.BytesIO(part.inline_data.data))
-          return part.inline_data.data
+            #image = PIL.Image.open(io.BytesIO(part.inline_data.data))
+            return part.inline_data.data
 
 
+''''
 # 使用 Hugging Face 生成圖片
 def generate_image_hf(prompt):
     API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
@@ -137,7 +134,7 @@ def generate_image_hf(prompt):
     else:
         print("圖片生成失敗:", response.text)
         return None
-    
+''''
 #==========================
 # 使用者上傳圖片
 #==========================