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()
|