Hugging Face's logo

Spaces:
mbar0075
/
YOLO-Application-Toolkit
Running

App Files Community
mbar0075 commited on
Commit
b95a330
·
verified ·
1 Parent(s): 5577900

Changed pizza models

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -40,10 +40,12 @@ MODELS = {
40
  "Smoke Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_smoke_detection.pt"),
41
  "Blood Cell Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_blood_cells.pt"),
42
  "Coins Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_coins.pt"),
43
- "Pizza Toppings Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_pizza.pt"),
44
  "Aquarium Fish Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_aquarium_fish.pt"),
45
  "Pelvis X-ray Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_pelvis_xray.pt"),
46
  "Road Signs Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_road_signs.pt"),
 
 
47
  }
48
 
49
 
@@ -59,10 +61,12 @@ EXAMPLE_DICT = {
59
  "Smoke Detection (YOLO11m)": example_dir + "7.jpg",
60
  "Blood Cell Detection (YOLO11m)": example_dir + "8.jpg",
61
  "Coins Detection (YOLO11m)": example_dir + "9.jpg",
62
- "Pizza Toppings Detection (YOLO11m)": example_dir + "10.jpg",
63
  "Aquarium Fish Detection (YOLO11m)": example_dir + "11.jpg",
64
  "Pelvis X-ray Detection (YOLO11m)": example_dir + "12.jpg",
65
  "Road Signs Detection (YOLO11m)": example_dir + "13.jpg",
 
 
66
  }
67
 
68
  LABEL_ANNOTATORS = sv.LabelAnnotator()
 
40
  "Smoke Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_smoke_detection.pt"),
41
  "Blood Cell Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_blood_cells.pt"),
42
  "Coins Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_coins.pt"),
43
+ "Pizza Toppings Detection (YOLOv8s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolov8s_pizza.pt"),
44
  "Aquarium Fish Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_aquarium_fish.pt"),
45
  "Pelvis X-ray Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_pelvis_xray.pt"),
46
  "Road Signs Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_road_signs.pt"),
47
+ "Pizza Toppings Detection (YOLO11m)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolo11m_pizza.pt"),
48
+ "Pizza Toppings Detection (YOLOv5s)": YOLO("https://huggingface.co/mbar0075/YOLO-Application-Toolkit/resolve/main/yolov5s_pizza.pt"),
49
  }
50
 
51
 
 
61
  "Smoke Detection (YOLO11m)": example_dir + "7.jpg",
62
  "Blood Cell Detection (YOLO11m)": example_dir + "8.jpg",
63
  "Coins Detection (YOLO11m)": example_dir + "9.jpg",
64
+ "Pizza Toppings Detection (YOLOv8s)": example_dir + "10.jpg",
65
  "Aquarium Fish Detection (YOLO11m)": example_dir + "11.jpg",
66
  "Pelvis X-ray Detection (YOLO11m)": example_dir + "12.jpg",
67
  "Road Signs Detection (YOLO11m)": example_dir + "13.jpg",
68
+ "Pizza Toppings Detection (YOLO11m)": example_dir + "10.jpg",
69
+ "Pizza Toppings Detection (YOLOv5s)": example_dir + "10.jpg",
70
  }
71
 
72
  LABEL_ANNOTATORS = sv.LabelAnnotator()