Spaces:
Sleeping
Sleeping
Update api_server.py
Browse files- 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 |
-
|
186 |
-
|
187 |
-
|
188 |
-
|
189 |
-
|
190 |
-
|
191 |
-
|
192 |
-
|
193 |
-
|
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,
|