Spaces:
Sleeping
Sleeping
Update tasks/image.py
Browse files- tasks/image.py +1 -1
tasks/image.py
CHANGED
@@ -128,7 +128,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
|
|
128 |
true_boxes_list.append([]) # Add empty list for no ground truth smoke
|
129 |
|
130 |
# Perform YOLO inference
|
131 |
-
results =
|
132 |
|
133 |
# Extract predicted box if predictions exist
|
134 |
if len(results[0].boxes):
|
|
|
128 |
true_boxes_list.append([]) # Add empty list for no ground truth smoke
|
129 |
|
130 |
# Perform YOLO inference
|
131 |
+
results = yolo_model .predict(image)
|
132 |
|
133 |
# Extract predicted box if predictions exist
|
134 |
if len(results[0].boxes):
|