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="yolov5n", device="cpu"):
+    def __init__(self, model_key="yolov8n", device="cpu"):
         """
         Initialize the Object Detection model using Ultralytics YOLO registry.
 
@@ -18,8 +18,6 @@ class ObjectDetector:
             device (str): 'cpu' or 'cuda'
         """
         alias_map = {
-            "yolov5n-seg": "yolov5n",
-            "yolov5s-seg": "yolov5s",
             "yolov8s": "yolov8s",
             "yolov8l": "yolov8l",
             "yolov11b": "yolov11b",