Update api_server.py

api_server.py

@@ -181,7 +181,7 @@ def predict():
             for yolo_img in yolo_file: # 每張切圖yolo_img
                 print("***** FLASK API---/predict 4. START CLIP *****")
                 
-                clip_model(choice,yolo_img,word)
+                clip_result = clip_model(choice,yolo_img,word)
                 
                 top_k_words.append(clip_result[0]) # CLIP預測3個結果(top_k_words)
                 encoded_images.append(image_to_base64(yolo_img))
@@ -218,7 +218,7 @@ def text2img():
     choice = request.form.get('choice')
     word = request.form.get('word')
     
-    clip_model(choice,None,word)
+    clip_result = clip_model(choice,None,word)
     
     print(f"===== FLASK API---/text2img 文字轉圖片result:{clip_result} =====")
     result_img = clip_result[2] # 已經是base64 coded