levydaniel commited on
Commit
2905ebd
·
verified ·
1 Parent(s): 243560a

Update tasks/image.py

Browse files
Files changed (1) hide show
  1. tasks/image.py +1 -3
tasks/image.py CHANGED
@@ -152,10 +152,8 @@ def compute_max_iou(true_boxes, pred_box):
152
  max_iou = max(max_iou, iou)
153
  return max_iou
154
 
155
- # @router.post(ROUTE, tags=["Image Task"],
156
- # description=DESCRIPTION)
157
  async def evaluate_image(model_path: str = "models/camera_models/yolo11n_default.engine",request: ImageEvaluationRequest = ImageEvaluationRequest()):
158
- # def evaluate_image(model_path: str, request: ImageEvaluationRequest = ImageEvaluationRequest()):
159
  """
160
  Evaluate image classification and object detection for forest fire smoke.
161
 
 
152
  max_iou = max(max_iou, iou)
153
  return max_iou
154
 
155
+ @router.post(ROUTE, tags=["Image Task"], description=DESCRIPTION)
 
156
  async def evaluate_image(model_path: str = "models/camera_models/yolo11n_default.engine",request: ImageEvaluationRequest = ImageEvaluationRequest()):
 
157
  """
158
  Evaluate image classification and object detection for forest fire smoke.
159