Update app.py

app.py

@@ -1,22 +1,23 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the text extraction model
-ocr_model = pipeline("image-to-text", model="microsoft/monologg-ocr")
+# Load the OCR model
+ocr_model = pipeline("image-to-text", model="microsoft/trocr-large-printed")
 
-def extract_text(image):
-    # Perform OCR
-    text = ocr_model(image)[0]['generated_text']
-    return text
+def recognize_text(image):
+    # Use the model on the input image
+    result = ocr_model(image)
+    return result[0]['generated_text']  # Return the recognized text
 
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=extract_text,
-    inputs=gr.Image(type="pil"),  # Accepts image input
-    outputs="text",                # Outputs extracted text
-    title="Image to Text Extractor",
-    description="Upload an image to extract text from it."
+# Set up the Gradio interface
+interface = gr.Interface(
+    fn=recognize_text,
+    inputs=gr.Image(type="filepath"),  # Use filepath to accept image input
+    outputs="text",  # Output the recognized text
+    title="OCR with Trocr",
+    description="Upload an image to recognize text using the Trocr model."
 )
 
-# Launch the interface
-iface.launch()
+# Launch the app
+if __name__ == "__main__":
+    interface.launch()