ning8429 commited on
Commit
1bc3888
·
verified ·
1 Parent(s): fdc68c3

Update api_server.py

Browse files
Files changed (1) hide show
  1. api_server.py +5 -1
api_server.py CHANGED
@@ -185,7 +185,11 @@ def predict():
185
  top_k_words.append(clip_result) # CLIP預測3個結果(top_k_words)
186
  encoded_images.append(image_to_base64(yolo_img))
187
  element_list.append(element)
188
- print(f"===== FLASK API---/predict CLIP result:{top_k_words} =====\n")
 
 
 
 
189
 
190
  # 建立回應資料
191
  response_data = {
 
185
  top_k_words.append(clip_result) # CLIP預測3個結果(top_k_words)
186
  encoded_images.append(image_to_base64(yolo_img))
187
  element_list.append(element)
188
+ print(f"===== FLASK API---/predict CLIP RESULT:{top_k_words} =====\n")
189
+
190
+ # 刪除已處理的圖片文件
191
+ print(f"===== FLASK API---/predict DELETE yolo_img:{yolo_img} =====\n")
192
+ os.remove(yolo_img)
193
 
194
  # 建立回應資料
195
  response_data = {