Hugging Face's logo

Spaces:
mbar0075
/
YOLO-Application-Toolkit
Running

App Files Community
mbar0075 commited on
Commit
2b0232f
·
verified ·
1 Parent(s): f47b425

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -55,6 +55,7 @@ MODELS = {
55
  "Car Parts Segmentation (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_seg_car_parts.pt"),
56
  "YOLO11s-Pose (COCO128)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s-pose.pt"),
57
  "Hand Keypoints Detection (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_hand_keypoints.pt"),
 
58
  }
59
 
60
 
@@ -85,6 +86,7 @@ EXAMPLE_DICT = {
85
  "Car Parts Segmentation (YOLO11s)": example_dir + "20.jpg",
86
  "YOLO11s-Pose (COCO128)": example_dir + "21.jpg",
87
  "Hand Keypoints Detection (YOLO11s)": example_dir + "22.jpg",
 
88
  }
89
 
90
  LABEL_ANNOTATORS = sv.LabelAnnotator()
 
55
  "Car Parts Segmentation (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_seg_car_parts.pt"),
56
  "YOLO11s-Pose (COCO128)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s-pose.pt"),
57
  "Hand Keypoints Detection (YOLO11s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11s_hand_keypoints.pt"),
58
+ "Garbage Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_garbage_detection.pt"),
59
  }
60
 
61
 
 
86
  "Car Parts Segmentation (YOLO11s)": example_dir + "20.jpg",
87
  "YOLO11s-Pose (COCO128)": example_dir + "21.jpg",
88
  "Hand Keypoints Detection (YOLO11s)": example_dir + "22.jpg",
89
+ "Garbage Detection (YOLO11m)": example_dir + "23.jpg",
90
  }
91
 
92
  LABEL_ANNOTATORS = sv.LabelAnnotator()