Hugging Face's logo

Spaces:
SEA-AI
/
VicAndTheBoys
Runtime error

App Files Community
VascoDVRodrigues commited on
Commit
8b16e95
·
1 Parent(s): bf31541

reqs

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -9,7 +9,7 @@ from yolov5.utils.torch_utils import select_device
9
  device = select_device('')
10
 
11
  # Load YOLOv5 model
12
- weights_path = './model/yolov5n6_RGB_D2304-v1_9C.pt'
13
  model = attempt_load(weights_path, map_location=device)
14
  stride = int(model.stride.max()) # model stride
15
 
 
9
  device = select_device('')
10
 
11
  # Load YOLOv5 model
12
+ weights_path = 'model/yolov5n6_RGB_D2304-v1_9C.pt'
13
  model = attempt_load(weights_path, map_location=device)
14
  stride = int(model.stride.max()) # model stride
15