Hugging Face's logo

Spaces:
DurgaDeepak
/
UVIS
Runtime error

App Files Community
DurgaDeepak commited on
Commit
330fb6c
·
verified ·
1 Parent(s): d61f1b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ logger = logging.getLogger(__name__)
32
 
33
  # Model mappings
34
  DETECTION_MODEL_MAP = {
35
- "YOLOv5-Nano": "yolov5n-seg",
36
- "YOLOv5-Small": "yolov5s-seg",
37
  "YOLOv8-Small": "yolov8s",
38
  "YOLOv8-Large": "yolov8l",
39
  "RT-DETR": "rtdetr" # For future support
 
32
 
33
  # Model mappings
34
  DETECTION_MODEL_MAP = {
35
+ "YOLOv5-Nano": "yolov5n",
36
+ "YOLOv5-Small": "yolov5s",
37
  "YOLOv8-Small": "yolov8s",
38
  "YOLOv8-Large": "yolov8l",
39
  "RT-DETR": "rtdetr" # For future support