Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import pipeline
 import re
 
-# Load a publicly available OCR pipeline (make sure this model exists)
-ocr_pipeline = pipeline("image-to-text", model="microsoft/vision-ocr")
+# Load the OCR pipeline
+ocr_pipeline = pipeline("image-to-text", model="microsoft/trocr-base-stage1")
 
 def perform_ocr(image):
     text = ocr_pipeline(image)[0]['generated_text']