ning8429 commited on
Commit
2e127b7
·
verified ·
1 Parent(s): 83dfc8d

Update api_server.py

Browse files
Files changed (1) hide show
  1. api_server.py +11 -11
api_server.py CHANGED
@@ -182,17 +182,17 @@ def predict():
182
  # encoded_images.append(image_to_base64(yolo_path))
183
 
184
  # 建立回應資料
185
- # response_data = {
186
- # 'message_id': message_id,
187
- # 'description': element_list,
188
- # 'images': [
189
- # {
190
- # 'encoded_image': encoded_image,
191
- # 'description_list': top_k_words
192
- # }
193
- # for encoded_image, elements in zip(encoded_images, element_list)
194
- # ]
195
- # }
196
  # response_data = {
197
  # 'message_id': message_id,
198
  # 'images': encoded_images,
 
182
  # encoded_images.append(image_to_base64(yolo_path))
183
 
184
  # 建立回應資料
185
+ response_data = {
186
+ 'message_id': message_id,
187
+ 'description': element_list,
188
+ 'images': [
189
+ {
190
+ 'encoded_image': encoded_image,
191
+ 'description_list': top_k_words
192
+ }
193
+ for encoded_image, description_list in zip(encoded_images, top_k_words)
194
+ ]
195
+ }
196
  # response_data = {
197
  # 'message_id': message_id,
198
  # 'images': encoded_images,