priyank-m commited on
Commit
e6a3630
·
1 Parent(s): 232d55e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
2
  import gradio as gr
3
 
4
- model=VisionEncoderDecoderModel.from_pretrained("priyank-m/beta-OCR")
5
  tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
6
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
7
 
 
1
  from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
2
  import gradio as gr
3
 
4
+ model=VisionEncoderDecoderModel.from_pretrained("priyank-m/mOCR")
5
  tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
6
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
7