randomshit11 commited on
Commit
acb9be1
·
verified ·
1 Parent(s): 2c806a3

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(r'E:\CV-project\MAIN-DATA\GRADIO\Yolo-v8-Video\yolov8x-seg.pt').to(device)
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').to(device)
12
 
13
  def process_video(input_video_path):
14
  cap = cv2.VideoCapture(input_video_path)