Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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('/
|
21 |
-
path='
|
22 |
-
source='
|
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 |
|