Vombit commited on
Commit
807b1cc
·
verified ·
1 Parent(s): 6cad3e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from PIL import Image
3
  from ultralytics import YOLO
4
  from ultralytics.utils.plotting import Annotator
5
 
6
- model_path = 'yolov10m_cs2.pt'
7
  model = YOLO(model_path, task="detect")
8
 
9
  def predict(image):
 
3
  from ultralytics import YOLO
4
  from ultralytics.utils.plotting import Annotator
5
 
6
+ model_path = 'yolo11n_cs2.pt'
7
  model = YOLO(model_path, task="detect")
8
 
9
  def predict(image):