Update models/detection/detector.py

models/detection/detector.py

@@ -9,7 +9,7 @@ import shutil
 logger = logging.getLogger(__name__)
 shutil.rmtree("models/detection/weights", ignore_errors=True)
 class ObjectDetector:
-    def __init__(self, model_key="yolov8n", device="cpu"):
+    def __init__(self, model_key="yolov8n.pt", device="cpu"):
         """
         Initialize the Object Detection model using Ultralytics YOLO registry.