kedimestan commited on
Commit
8ffa231
1 Parent(s): ee049fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -6,14 +6,14 @@ from ultralyticsplus import YOLO
6
  # Download images
7
 
8
  model_names = [
9
- "yolov8n-seg.pt",
10
- "yolov8s-seg.pt",
11
- "yolov8m-seg.pt",
12
- "yolov8l-seg.pt",
13
- "yolov8x-seg.pt",
14
  ]
15
 
16
- current_model_name = "yolov8m-seg.pt"
17
  model = YOLO(current_model_name)
18
 
19
 
 
6
  # Download images
7
 
8
  model_names = [
9
+ "yolov8n.pt",
10
+ "yolov8s.pt",
11
+ "yolov8m.pt",
12
+ "yolov8l.pt",
13
+ "yolov8x.pt",
14
  ]
15
 
16
+ current_model_name = "yolov8m.pt"
17
  model = YOLO(current_model_name)
18
 
19