Spaces:

RivianG
/

Asis

Runtime error

RivianG commited on Mar 8, 2022

Commit

9612063

•

1 Parent(s): 014f3ef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import os
 import streamlit as st
-model = torch.hub.load('/home/ai/Downloads/yolov5', 'custom', path='/home/ai/Desktop/Icevision/yoloocr_best.engine', source='local')
 model.cuda()
 model.conf = 0.5
 license = DetectLicensePlate()

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