ning8429 commited on
Commit
27d4de9
·
verified ·
1 Parent(s): 43498e1

Update api_server.py

Browse files
Files changed (1) hide show
  1. api_server.py +5 -4
api_server.py CHANGED
@@ -96,7 +96,7 @@ check_memory_usage()
96
  # Initialize the Flask application
97
  app = Flask(__name__)
98
  # Initialize the ClipModel at the start
99
- clip_model = ClipModel()
100
 
101
  check_memory_usage()
102
 
@@ -158,9 +158,10 @@ def predict():
158
  element_list.append(element)
159
 
160
  for yolo_img in yolo_file: # 每張切圖yolo_img
161
- top_k_words.append(clip_model.clip_result(yolo_img)) # CLIP預測3個結果(top_k_words)
162
- encoded_images.append(image_to_base64(yolo_img))
163
- print(f"**{yolo_img}:{top_k_words}**\n")
 
164
 
165
  # if element_counts[element] > 1: #某隻角色的數量>1
166
  # yolo_path = f"{YOLO_DIR}/{message_id}/{element}"
 
96
  # Initialize the Flask application
97
  app = Flask(__name__)
98
  # Initialize the ClipModel at the start
99
+ # clip_model = ClipModel()
100
 
101
  check_memory_usage()
102
 
 
158
  element_list.append(element)
159
 
160
  for yolo_img in yolo_file: # 每張切圖yolo_img
161
+ print("yolo _img:", yolo_img)
162
+ #top_k_words.append(clip_model.clip_result(yolo_img)) # CLIP預測3個結果(top_k_words)
163
+ #encoded_images.append(image_to_base64(yolo_img))
164
+ #print(f"**{yolo_img}:{top_k_words}**\n")
165
 
166
  # if element_counts[element] > 1: #某隻角色的數量>1
167
  # yolo_path = f"{YOLO_DIR}/{message_id}/{element}"