randomshit11 commited on
Commit
5526a08
·
verified ·
1 Parent(s): 06b0435

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from ultralytics import YOLO
8
  # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
 
10
  # Load the YOLOv8 model
11
- model = YOLO('best.pt')
12
 
13
  def process_video(input_video_path):
14
  cap = cv2.VideoCapture(input_video_path)
 
8
  # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
 
10
  # Load the YOLOv8 model
11
+ model = YOLO('yolov8x-seg.pt')
12
 
13
  def process_video(input_video_path):
14
  cap = cv2.VideoCapture(input_video_path)