Spaces:
Build error
Build error
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,
|