Spaces:

RivianG
/

Asis

Runtime error

RivianG commited on Mar 8, 2022

Commit

4313318

1 Parent(s): e338936

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import streamlit as st
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='yoloocr_best.pt')
-model.cuda()
 model.conf = 0.5
 license = DetectLicensePlate()
 counter  = dict()

 model = torch.hub.load('ultralytics/yolov5', 'custom', path='yoloocr_best.pt')
+model.cpu()
 model.conf = 0.5
 license = DetectLicensePlate()
 counter  = dict()