Update app.py
Browse files
app.py
CHANGED
|
@@ -29,6 +29,7 @@ image_file = st.file_uploader("Upload Images", type=["png","jpg","jpeg"])
|
|
| 29 |
submit = st.button('Generate')
|
| 30 |
|
| 31 |
if submit:
|
|
|
|
| 32 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='yoloocr_best.pt')
|
| 33 |
model.cpu()
|
| 34 |
model.conf = 0.5
|
|
|
|
| 29 |
submit = st.button('Generate')
|
| 30 |
|
| 31 |
if submit:
|
| 32 |
+
image = load_image(image_file)
|
| 33 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='yoloocr_best.pt')
|
| 34 |
model.cpu()
|
| 35 |
model.conf = 0.5
|