Spaces:

Prevantec
/

platerecognition

Runtime error

xangcastle commited on Dec 10, 2022

Commit

3f7b316

1 Parent(s): ff3c638

activate cpu

Files changed (1) hide show

detector/utils.py CHANGED Viewed

@@ -10,7 +10,7 @@ BASE_DIR = os.path.abspath(os.getcwd())
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 cpu = torch.device('cpu')
-print('Loading models...', device)
 model_plates = torch.hub.load('ultralytics/yolov5', 'custom',
                               path=os.path.join(BASE_DIR, 'detector', 'static', 'plates.pt'), device=cpu)

 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 cpu = torch.device('cpu')
+print('Loading models...', cpu)
 model_plates = torch.hub.load('ultralytics/yolov5', 'custom',
                               path=os.path.join(BASE_DIR, 'detector', 'static', 'plates.pt'), device=cpu)