Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ submit = st.button('Generate')
|
|
30 |
|
31 |
if submit:
|
32 |
image = load_image(image_file)
|
33 |
-
model = torch.hub.load('
|
34 |
model.cpu()
|
35 |
model.conf = 0.5
|
36 |
license = DetectLicensePlate()
|
|
|
30 |
|
31 |
if submit:
|
32 |
image = load_image(image_file)
|
33 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='yoloocrv2_1.pt')
|
34 |
model.cpu()
|
35 |
model.conf = 0.5
|
36 |
license = DetectLicensePlate()
|