Spaces:

DurgaDeepak
/

UVIS

Runtime error

DurgaDeepak commited on May 22

Commit

7826054

verified ·

1 Parent(s): b44a48c

Update core/process.py

Files changed (1) hide show

core/process.py CHANGED Viewed

@@ -18,12 +18,10 @@ logger = logging.getLogger(__name__)
 # Model mappings
 DETECTION_MODEL_MAP = {
-    "YOLOv5-Nano": "yolov5n",
-    "YOLOv5-Small": "yolov5s",
     "YOLOv8-Small": "yolov8s",
     "YOLOv8-Large": "yolov8l",
-    "YOLOv11-Beta": "yolov11b",
-    "RT-DETR": "rtdetr"
 }
 SEGMENTATION_MODEL_MAP = {

 # Model mappings
 DETECTION_MODEL_MAP = {
+    "YOLOv8-Nano": "yolov8n",
     "YOLOv8-Small": "yolov8s",
     "YOLOv8-Large": "yolov8l",
+    "YOLOv11-Beta": "yolov11b"
 }
 SEGMENTATION_MODEL_MAP = {