priyank-m commited on
Commit
6607a8c
1 Parent(s): ce15747

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,12 +2,12 @@ from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecode
2
  import gradio as gr
3
 
4
  title="Multilingual OCR (currently recognises: English and Chinese)"
5
- description="mOCR(multilingual-OCR) is a Vision-Encoder-Decoder model which uses pre-trained facebook's vit-mae-large as the encoder and xlm-roberta-base as the decoder. \nIt has been trained on IAM, SROIE 2019, text_renderer Chinese (synthetic) and TRDG (synthetic) datasets (amounting to approx 1.4 Million samples) for English and Chinese document text-recognition."
6
  examples =[["demo_image/img1.png"], ["demo_image/img2.jpeg"], ["demo_image/img3.jpeg"], ["demo_image/img4.jpeg"], ["demo_image/img5.jpeg"], ["demo_image/img6.jpeg"]]
7
 
8
 
9
 
10
- model=VisionEncoderDecoderModel.from_pretrained("priyank-m/mOCR")
11
  tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
12
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
13
 
 
2
  import gradio as gr
3
 
4
  title="Multilingual OCR (currently recognises: English and Chinese)"
5
+ description="m_OCR(multilingual OCR) is a Vision-Encoder-Decoder model which uses pre-trained facebook's vit-mae-large as the encoder and xlm-roberta-base as the decoder. \nIt has been trained on IAM, SROIE 2019, text_renderer Chinese (synthetic) and TRDG (synthetic) datasets (amounting to approx 1.4 Million samples) for English and Chinese document text-recognition."
6
  examples =[["demo_image/img1.png"], ["demo_image/img2.jpeg"], ["demo_image/img3.jpeg"], ["demo_image/img4.jpeg"], ["demo_image/img5.jpeg"], ["demo_image/img6.jpeg"]]
7
 
8
 
9
 
10
+ model=VisionEncoderDecoderModel.from_pretrained("priyank-m/m_OCR")
11
  tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
12
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
13