asif00 commited on
Commit
10c41bf
·
1 Parent(s): ce7f5f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from ultralytics import YOLO
5
 
6
 
7
  def load_yolo_model():
8
-     return YOLO()
9
 
10
 
11
  def process_video(model, cap):
 
5
 
6
 
7
  def load_yolo_model():
8
+     return YOLO('yolov8n-seg.pt')
9
 
10
 
11
  def process_video(model, cap):