Update core/process.py

core/process.py

@@ -18,16 +18,17 @@ logger = logging.getLogger(__name__)
 
 # Model mappings
 DETECTION_MODEL_MAP = {
-    "YOLOv5-Nano": "yolov5n-seg",
-    "YOLOv5-Small": "yolov5s-seg",
+    "YOLOv5-Nano": "yolov5n",
+    "YOLOv5-Small": "yolov5s",
     "YOLOv8-Small": "yolov8s",
     "YOLOv8-Large": "yolov8l",
-    "RT-DETR": "rtdetr"  # For future support
+    "YOLOv11-Beta": "yolov11b",
+    "RT-DETR": "rtdetr"
 }
 
 SEGMENTATION_MODEL_MAP = {
-    "SegFormer-B0": "nvidia/segformer-b0-finetuned-ade-512-512",
-    "SegFormer-B5": "nvidia/segformer-b5-finetuned-ade-512-512",
+    "SegFormer-B0": "segformer_b0",
+    "SegFormer-B5": "segformer_b5",
     "DeepLabV3-ResNet50": "deeplabv3_resnet50"
 }
 
@@ -39,6 +40,7 @@ DEPTH_MODEL_MAP = {
     "DPT Beit Large 512": "dpt_beit_large_512"
 }
 
+
 @timeout_decorator.timeout(35, use_signals=False)  # 35 sec limit per image
 def process_image(
     image: Image.Image,