Spaces:
Sleeping
Sleeping
Update api_server.py
Browse files- api_server.py +1 -1
api_server.py
CHANGED
@@ -80,7 +80,7 @@ def predict():
|
|
80 |
# 儲存辨識後的圖片到指定資料夾
|
81 |
for result in results:
|
82 |
# 保存完整圖片
|
83 |
-
yolo_path = YOLO_DIR
|
84 |
result.save(yolo_path)
|
85 |
saved_images.append(yolo_path)
|
86 |
|
|
|
80 |
# 儲存辨識後的圖片到指定資料夾
|
81 |
for result in results:
|
82 |
# 保存完整圖片
|
83 |
+
yolo_path = f'{YOLO_DIR}/results_{Path(result.path).name}'
|
84 |
result.save(yolo_path)
|
85 |
saved_images.append(yolo_path)
|
86 |
|