Hugging Face's logo

Spaces:
DurgaDeepak
/
UVIS
Runtime error

App Files Community
DurgaDeepak commited on
Commit
b44a48c
·
verified ·
1 Parent(s): 9d9a80a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -39,12 +39,13 @@ logger = logging.getLogger(__name__)
39
 
40
  # Model mappings
41
  DETECTION_MODEL_MAP = {
42
- "YOLOv8-Nano": "yolov8n.pt",
43
- "YOLOv8-Small": "yolov8s.pt",
44
- "YOLOv8-Large": "yolov8l.pt",
45
- "YOLOv11-Beta": "yolov11b.pt"
46
  }
47
 
 
48
  SEGMENTATION_MODEL_MAP = {
49
  "SegFormer-B0": "segformer_b0",
50
  "SegFormer-B5": "segformer_b5",
 
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
+
49
  SEGMENTATION_MODEL_MAP = {
50
  "SegFormer-B0": "segformer_b0",
51
  "SegFormer-B5": "segformer_b5",