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-handwritten")
-model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwritten").to(device)
+processor = TrOCRProcessor.from_pretrained("MQAnh/my-awesome-model")
+model = VisionEncoderDecoderModel.from_pretrained("MQAnh/my-awesome-model").to(device)
 
 def ocr_infer(image):
     pixel_values = processor(images=image, return_tensors="pt").pixel_values.to(device)
@@ -20,7 +20,7 @@ iface = gr.Interface(
     fn=ocr_infer,
     inputs=gr.Image(type="pil"),
     outputs="text",
-    title="Image to Text (OCR) ver6",
+    title="Image to Text (OCR) ver7",
     description="Upload a handwritten or printed image to extract text using TrOCR."
 )