tom-b974 commited on
Commit
d92f5e3
·
verified ·
1 Parent(s): 1e16d37

Update tasks/image.py

Browse files
Files changed (1) hide show
  1. 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 = model.predict(image)
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):