Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoProcessor, VisionEncoderDecoderModel
+from transformers import AutoTokenizer, AutoProcessor, VisionEncoderDecoderModel, TrOCRProcessor
 from vllm import LLM, SamplingParams
 from PIL import Image
 
@@ -13,7 +13,8 @@ vllm_model = LLM(model=model_name, tensor_parallel_size=1, device="cpu")
 # Load the OCR model and processor
 ocr_model_name = "microsoft/trocr-small-handwritten"
 ocr_model = VisionEncoderDecoderModel.from_pretrained(ocr_model_name)
-ocr_processor = AutoProcessor.from_pretrained(ocr_model_name)
+ocr_processor = TrOCRProcessor.from_pretrained(ocr_model_name)
+#ocr_processor = AutoProcessor.from_pretrained(ocr_model_name)
 
 def generate_response(prompt, max_tokens, temperature, top_p):
     # Define sampling parameters