Update app.py
Browse files
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()
|