Hugging Face's logo

Spaces:
DurgaDeepak
/
UVIS
Runtime error

App Files Community
DurgaDeepak commited on
Commit
fddc40c
·
verified ·
1 Parent(s): 84fb7e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -39,12 +39,10 @@ logger = logging.getLogger(__name__)
39
 
40
  # Model mappings
41
  DETECTION_MODEL_MAP = {
42
- "YOLOv5-Nano": "yolov5n",
43
- "YOLOv5-Small": "yolov5s",
44
  "YOLOv8-Small": "yolov8s",
45
  "YOLOv8-Large": "yolov8l",
46
- "YOLOv11-Beta": "yolov11b",
47
- "RT-DETR": "rtdetr"
48
  }
49
 
50
  SEGMENTATION_MODEL_MAP = {
 
39
 
40
  # Model mappings
41
  DETECTION_MODEL_MAP = {
42
+ "YOLOv8-Nano": "yolov8n",
 
43
  "YOLOv8-Small": "yolov8s",
44
  "YOLOv8-Large": "yolov8l",
45
+ "YOLOv11-Beta": "yolov11b"
 
46
  }
47
 
48
  SEGMENTATION_MODEL_MAP = {