Lljedwards commited on
Commit
d150404
·
verified ·
1 Parent(s): 5a8f34d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -17,9 +17,9 @@ mp_drawing = mp.solutions.drawing_utils
17
 
18
  # Load your local YOLOv5 model for javelin detection.
19
  # Adjust these paths as needed.
20
- model = torch.hub.load('/Users/user/yolov5', 'custom',
21
- path='/Users/user/yolov5/runs/train/exp/weights/best.pt',
22
- source='local')
23
  model.conf = 0.5 # detection confidence threshold
24
 
25
 
 
17
 
18
  # Load your local YOLOv5 model for javelin detection.
19
  # Adjust these paths as needed.
20
+ model = torch.hub.load('ultralytics/yolov5', 'custom',
21
+ path='weights/best.pt', # relative path in your repo
22
+ source='github')
23
  model.conf = 0.5 # detection confidence threshold
24
 
25