vteam27
commited on
Commit
·
5fee3bf
1
Parent(s):
2c62fa3
Changed predictor model
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from doctr.models import ocr_predictor
|
|
| 6 |
import gradio as gr
|
| 7 |
from PIL import Image
|
| 8 |
|
| 9 |
-
predictor = ocr_predictor(pretrained=True)
|
| 10 |
|
| 11 |
title="DocTR OCR (PDL Demo)"
|
| 12 |
description="Upload an image to get the OCR results !"
|
|
|
|
| 6 |
import gradio as gr
|
| 7 |
from PIL import Image
|
| 8 |
|
| 9 |
+
predictor = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn',pretrained=True)
|
| 10 |
|
| 11 |
title="DocTR OCR (PDL Demo)"
|
| 12 |
description="Upload an image to get the OCR results !"
|