Spaces:

DurgaDeepak
/

UVIS

Running on Zero

DurgaDeepak commited on May 21

Commit

0c0955d

verified ·

1 Parent(s): 5b19bda

Update models/detection/detector.py

Files changed (1) hide show

models/detection/detector.py CHANGED Viewed

@@ -29,7 +29,7 @@ class ObjectDetector:
             "yolov11b": "yolov11b"
         }
-        resolved_key = alias_map.get(model_key.lower(), model_key.lower())
         # HF repo map
         hf_map = {

             "yolov11b": "yolov11b"
         }
+        resolved_key = model_key.lower().replace(".pt", "")
         # HF repo map
         hf_map = {