hantech commited on
Commit
4733ddf
·
verified ·
1 Parent(s): d26bf91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,8 +9,7 @@ import base64
9
  from utils import HocrParser
10
 
11
  predictor = ocr_predictor(det_arch='db_mobilenet_v3_large', reco_arch='crnn_vgg16_bn',pretrained=True)
12
- print(predictor.cfg['vocab'])
13
- predictor.cfg['vocab']='vietnamese'
14
  title="DocTR OCR (PDL Demo)"
15
  description="Upload an image to get the OCR results !"
16
 
 
9
  from utils import HocrParser
10
 
11
  predictor = ocr_predictor(det_arch='db_mobilenet_v3_large', reco_arch='crnn_vgg16_bn',pretrained=True)
12
+ predictor.reco_predictor.model.cfg['vocab']='0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!”#$%&’()*+,-./:;<=>?@[]^_`{|}~°£€¥¢฿áàảạãăắằẳẵặâấầẩẫậéèẻẽẹêếềểễệóòỏõọôốồổộỗơớờởợỡúùủũụưứừửữựiíìỉĩịýỳỷỹỵÁÀẢẠÃĂẮẰẲẴẶÂẤẦẨẪẬÉÈẺẼẸÊẾỀỂỄỆÓÒỎÕỌÔỐỒỔỘỖƠỚỜỞỢỠÚÙỦŨỤƯỨỪỬỮỰIÍÌỈĨỊÝỲỶỸỴ'
 
13
  title="DocTR OCR (PDL Demo)"
14
  description="Upload an image to get the OCR results !"
15