Update api_server.py

api_server.py

@@ -25,8 +25,8 @@ load_type = 'local'
 MODEL_YOLO = "yolo11_detect_best_241024_1.pt"
 MODEL_DIR = "./artifacts/models"
 YOLO_DIR = "./artifacts/yolo"
-IMG2TEXT_URL = "https://fd39e54bcb191a37bf.gradio.live/"
-TEXT2IMG_URL = "https://0816a20213827bdb75.gradio.live/"
+IMG2TEXT_URL = "https://fd39e54bcb191a37bf.gradio.live/" # 圖生文
+TEXT2IMG_URL = "https://e0ac675070fceb8e63.gradio.live/" # 文生圖
 
 
 # Load the saved YOLO model into memory
@@ -217,9 +217,8 @@ def text2img():
     		word= text_message,
     		api_name="/predict"
     )
-    print(f"===== FLASK API---/text2img 文字轉圖片result[0]:{result[0]} =====")
-    result_img = image_to_base64(result[0])
-    print(f"===== FLASK API---/text2img 文字轉圖片轉base64:{result_img} =====")
+    print(f"===== FLASK API---/text2img 文字轉圖片result:{result} =====")
+    result_img = result[2] # 已經是base64 coded
 
     # 建立回應資料
     response_data = {