assentian1970 commited on
Commit
8f0a5f4
·
verified ·
1 Parent(s): 0c13777

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import numpy as np
15
  import cv2
16
 
17
  # Add this after other model configurations
18
- YOLO_MODEL = YOLO('/teamspace/studios/this_studio/best_yolov11.pt') # Load YOLOv11 model
19
 
20
  # Check if CUDA is available
21
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
15
  import cv2
16
 
17
  # Add this after other model configurations
18
+ YOLO_MODEL = YOLO('best_yolov11.pt') # Load YOLOv11 model
19
 
20
  # Check if CUDA is available
21
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"