Spaces:

DurgaDeepak
/

UVIS

Running on Zero

DurgaDeepak commited on May 21

Commit

21f4ec4

verified ·

1 Parent(s): a0ce3ec

Update models/detection/detector.py

Files changed (1) hide show

models/detection/detector.py CHANGED Viewed

@@ -27,7 +27,7 @@ class ObjectDetector:
         resolved_key = alias_map.get(raw_key, raw_key)
         self.device = device
-        self.model = YOLO(resolved_key)
         logger.info(f" Ultralytics YOLO model '{resolved_key}' initialized on {device}")

         resolved_key = alias_map.get(raw_key, raw_key)
         self.device = device
+        self.model = YOLO(f"{resolved_key}.pt")
         logger.info(f" Ultralytics YOLO model '{resolved_key}' initialized on {device}")