Hugging Face's logo

Spaces:
DurgaDeepak
/
UVIS
Runtime error

App Files Community
DurgaDeepak commited on
Commit
27e0318
·
verified ·
1 Parent(s): 8601497

Update utils/model_downloader.py

Browse files
Files changed (1) hide show
  1. utils/model_downloader.py +1 -2
utils/model_downloader.py CHANGED
@@ -16,8 +16,7 @@ HF_MODELS = {
16
  "dpt_beit_large_512": ("isl-org/MiDaS", "dpt_beit_large_512.pt"),
17
 
18
  # Object Detection
19
- "yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
20
- "yolov5s": ("ultralytics/yolov5", "yolov5s.pt"),
21
  "yolov8s": ("ultralytics/yolov8", "yolov8s.pt"),
22
  "yolov8l": ("ultralytics/yolov8", "yolov8l.pt"),
23
  "yolov11b": ("Ultralytics/YOLO11", "yolov11b.pt"),
 
16
  "dpt_beit_large_512": ("isl-org/MiDaS", "dpt_beit_large_512.pt"),
17
 
18
  # Object Detection
19
+ "yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
 
20
  "yolov8s": ("ultralytics/yolov8", "yolov8s.pt"),
21
  "yolov8l": ("ultralytics/yolov8", "yolov8l.pt"),
22
  "yolov11b": ("Ultralytics/YOLO11", "yolov11b.pt"),