Update tasks/image.py

tasks/image.py

@@ -31,7 +31,7 @@ device = torch.device("cuda")
 
 def load_camera_models():
     models = {}
-    folder = "cameras_dataset/"
+    folder = "models/camera_models"
     cameras = ['brison-200', 'brison-110', 'courmettes-212', 'courmettes-160', 'brison-290', 'marguerite-29','default']
 
     # Ensure the folder exists
@@ -154,7 +154,7 @@ def compute_max_iou(true_boxes, pred_box):
 
 # @router.post(ROUTE, tags=["Image Task"],
 #              description=DESCRIPTION)
-async def evaluate_image(model_path,request: ImageEvaluationRequest = ImageEvaluationRequest()):
+async def evaluate_image(model_path: str = "models/camera_models/yolo11n_default.engine",request: ImageEvaluationRequest = ImageEvaluationRequest()):
 # def evaluate_image(model_path: str, request: ImageEvaluationRequest = ImageEvaluationRequest()):
     """
     Evaluate image classification and object detection for forest fire smoke.