Update app.py

app.py

@@ -1,9 +1,9 @@
-from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer
+from transformers import ViTFeatureExtractor, BertTokenizer, VisionEncoderDecoderModel, AutoTokenizer, AutoFeatureExtractor
 import gradio as gr
 
-model=VisionEncoderDecoderModel.from_pretrained("priyank-m/beit-roberta-OCR")
-tokenizer =  AutoTokenizer.from_pretrained("roberta-large")
-feature_extractor = ViTFeatureExtractor.from_pretrained("microsoft/beit-large-patch16-224")
+model=VisionEncoderDecoderModel.from_pretrained("priyank-m/beta-OCR")
+tokenizer =  AutoTokenizer.from_pretrained("xlm-roberta-base")
+feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/vit-mae-large")
 
 def run_ocr(image):
   pixel_values = feature_extractor(image, return_tensors="pt").pixel_values