Hugging Face's logo

Spaces:
DurgaDeepak
/
UVIS
Running on Zero

App Files Community
DurgaDeepak commited on
Commit
3340b09
·
verified ·
1 Parent(s): eb9703e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -36,12 +36,13 @@ DETECTION_MODEL_MAP = {
36
  "YOLOv5-Small": "yolov5s",
37
  "YOLOv8-Small": "yolov8s",
38
  "YOLOv8-Large": "yolov8l",
39
- "RT-DETR": "rtdetr" # For future support
 
40
  }
41
 
42
  SEGMENTATION_MODEL_MAP = {
43
- "SegFormer-B0": "nvidia/segformer-b0-finetuned-ade-512-512",
44
- "SegFormer-B5": "nvidia/segformer-b5-finetuned-ade-512-512",
45
  "DeepLabV3-ResNet50": "deeplabv3_resnet50"
46
  }
47
 
@@ -53,6 +54,7 @@ DEPTH_MODEL_MAP = {
53
  "DPT Beit Large 512": "dpt_beit_large_512"
54
  }
55
 
 
56
  # Resource Limits
57
  MAX_IMAGE_MB = 15
58
  MAX_IMAGE_RES = (1920, 1080)
 
36
  "YOLOv5-Small": "yolov5s",
37
  "YOLOv8-Small": "yolov8s",
38
  "YOLOv8-Large": "yolov8l",
39
+ "YOLOv11-Beta": "yolov11b",
40
+ "RT-DETR": "rtdetr"
41
  }
42
 
43
  SEGMENTATION_MODEL_MAP = {
44
+ "SegFormer-B0": "segformer_b0",
45
+ "SegFormer-B5": "segformer_b5",
46
  "DeepLabV3-ResNet50": "deeplabv3_resnet50"
47
  }
48
 
 
54
  "DPT Beit Large 512": "dpt_beit_large_512"
55
  }
56
 
57
+
58
  # Resource Limits
59
  MAX_IMAGE_MB = 15
60
  MAX_IMAGE_RES = (1920, 1080)