Update app.py

app.py

@@ -1,10 +1,13 @@
 import gradio as gr
-from transformers import VisionEncoderDecoderModel, TrOCRProcessor
+from transformers import VisionEncoderDecoderModel, TrOCRProcessor,  AutoTokenizer
 from PIL import Image
 
 # Load the model and processor from Hugging Face
+tokenizer = AutoTokenizer.from_pretrained("paudelanil/trocr-devanagari")
 model = VisionEncoderDecoderModel.from_pretrained("paudelanil/trocr-devanagari")
-processor = TrOCRProcessor.from_pretrained("paudelanil/trocr-devanagari")
+feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
+
+processor = TrOCRProcessor(feature_extractor=feature_extractor, tokenizer=tokenizer)
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model.to(device)