Spaces:
Runtime error
Runtime error
Update models/detection/detector.py
Browse files
models/detection/detector.py
CHANGED
|
@@ -28,8 +28,9 @@ class ObjectDetector:
|
|
| 28 |
"yolov11b": "yolov11b",
|
| 29 |
"rtdetr": "rtdetr"
|
| 30 |
}
|
| 31 |
-
|
| 32 |
-
|
|
|
|
| 33 |
repo_map = {
|
| 34 |
"yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
|
| 35 |
"yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
|
|
|
|
| 28 |
"yolov11b": "yolov11b",
|
| 29 |
"rtdetr": "rtdetr"
|
| 30 |
}
|
| 31 |
+
|
| 32 |
+
raw_key = model_key.lower()
|
| 33 |
+
model_key = alias_map.get(raw_key, raw_key) #to make model_key case-insensitive
|
| 34 |
repo_map = {
|
| 35 |
"yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
|
| 36 |
"yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
|