Spaces:
Runtime error
Runtime error
Update models/detection/detector.py
Browse files
models/detection/detector.py
CHANGED
|
@@ -35,6 +35,7 @@ class ObjectDetector:
|
|
| 35 |
model_key = alias_map.get(raw_key, raw_key) #to make model_key case-insensitive
|
| 36 |
repo_map = {
|
| 37 |
"yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
|
|
|
|
| 38 |
"yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
|
| 39 |
"yolov8s": ("ultralytics/yolov8", "yolov8s.pt"),
|
| 40 |
"yolov8l": ("ultralytics/yolov8", "yolov8l.pt"),
|
|
|
|
| 35 |
model_key = alias_map.get(raw_key, raw_key) #to make model_key case-insensitive
|
| 36 |
repo_map = {
|
| 37 |
"yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
|
| 38 |
+
"yolov5s": ("ultralytics/yolov5", "yolov5s.pt"),
|
| 39 |
"yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
|
| 40 |
"yolov8s": ("ultralytics/yolov8", "yolov8s.pt"),
|
| 41 |
"yolov8l": ("ultralytics/yolov8", "yolov8l.pt"),
|