DocTRv1

Build error

hantech commited on Jan 15, 2024

Commit

d26bf91

verified ·

1 Parent(s): b7413a6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ import base64
 from utils import HocrParser
 predictor = ocr_predictor(det_arch='db_mobilenet_v3_large', reco_arch='crnn_vgg16_bn',pretrained=True)
-print(predictor.model.cfg['vocab'])
-predictor.model.cfg['vocab']='vietnamese'
 title="DocTR OCR (PDL Demo)"
 description="Upload an image to get the OCR results !"

 from utils import HocrParser
 predictor = ocr_predictor(det_arch='db_mobilenet_v3_large', reco_arch='crnn_vgg16_bn',pretrained=True)
+print(predictor.cfg['vocab'])
+predictor.cfg['vocab']='vietnamese'
 title="DocTR OCR (PDL Demo)"
 description="Upload an image to get the OCR results !"