mbar0075 commited on
Commit
81bbc69
·
verified ·
1 Parent(s): 2459655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -48,6 +48,7 @@ MODELS = {
48
  "Pelvis X-ray Segmentation (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_seg_pelvis_xray.pt"),
49
  "Teeth X-ray Segmentation (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_seg_teeth_xray.pt"),
50
  "Football Players Detection (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_football_players.pt"),
 
51
  }
52
 
53
 
@@ -71,6 +72,7 @@ EXAMPLE_DICT = {
71
  "Pelvis X-ray Segmentation (YOLO11s)": example_dir + "12.jpg",
72
  "Teeth X-ray Segmentation (YOLO11s)": example_dir + "14.jpg",
73
  "Football Players Detection (YOLO11s)": example_dir + "15.jpg",
 
74
  }
75
 
76
  LABEL_ANNOTATORS = sv.LabelAnnotator()
 
48
  "Pelvis X-ray Segmentation (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_seg_pelvis_xray.pt"),
49
  "Teeth X-ray Segmentation (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_seg_teeth_xray.pt"),
50
  "Football Players Detection (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_football_players.pt"),
51
+ "Litter Detection (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_litter_detection.pt"),
52
  }
53
 
54
 
 
72
  "Pelvis X-ray Segmentation (YOLO11s)": example_dir + "12.jpg",
73
  "Teeth X-ray Segmentation (YOLO11s)": example_dir + "14.jpg",
74
  "Football Players Detection (YOLO11s)": example_dir + "15.jpg",
75
+ "Litter Detection (YOLO11s)": example_dir + "16.jpg",
76
  }
77
 
78
  LABEL_ANNOTATORS = sv.LabelAnnotator()