Update app.py

app.py

@@ -6,8 +6,8 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the model
-processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-stage1")
-model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-stage1").to(device)
+processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
+model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwritten").to(device)
 
 def ocr_infer(image):
     pixel_values = processor(images=image, return_tensors="pt").pixel_values.to(device)