Dawars commited on
Commit
a63b2c7
1 Parent(s): ee845ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,13 +1,13 @@
1
  from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
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 (based on the concept of TrOCR) 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
 
 
1
  from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
2
  import gradio as gr
3
 
4
+ title="Handwritten Phone Number OCR"
5
+ description="handwritten_phone_number_OCR is a Vision-Encoder-Decoder model (based on the concept of TrOCR) which uses pre-trained facebook's vit-mae-large as the encoder and xlm-roberta-base as the decoder. \nIt has been trained on MNIST (2M synthetic images)."
6
+ examples =[["demo_image/0313474611_lauf25.png"], ["demo_image/0466975865_nr4ywx.png"], ["demo_image/0473227403_7lnod1.png"], ["demo_image/0728880927_jr987p.png"], ["demo_image/0922853144_1o4ay5.png"]]
7
 
8
 
9
 
10
+ model=VisionEncoderDecoderModel.from_pretrained("dawars/handwritten_phone_number_OCR")
11
  tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
12
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
13