Spaces:

DurgaDeepak
/

UVIS

Running on Zero

DurgaDeepak commited on May 17

Commit

330fb6c

verified ·

1 Parent(s): d61f1b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,8 +32,8 @@ logger = logging.getLogger(__name__)
 # Model mappings
 DETECTION_MODEL_MAP = {
-    "YOLOv5-Nano": "yolov5n-seg",
-    "YOLOv5-Small": "yolov5s-seg",
     "YOLOv8-Small": "yolov8s",
     "YOLOv8-Large": "yolov8l",
     "RT-DETR": "rtdetr"  # For future support

 # Model mappings
 DETECTION_MODEL_MAP = {
+    "YOLOv5-Nano": "yolov5n",
+    "YOLOv5-Small": "yolov5s",
     "YOLOv8-Small": "yolov8s",
     "YOLOv8-Large": "yolov8l",
     "RT-DETR": "rtdetr"  # For future support