File size: 861 Bytes
8219e88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from transformers import pipeline
import gradio as gr
# Load the model
pipe = pipeline("image-to-text", model="jinhybr/OCR-Donut-CORD")
# Function to process the image and extract text
def extract_text(image):
# Pass the image to the pipeline
result = pipe(image)
# Return the text from the image
return result[0]['generated_text'] if result else "No text detected"
# Define the Gradio interface
iface = gr.Interface(
fn=extract_text, # The function that processes the image
inputs=gr.Image(type="pil"), # Input is an image (PIL format)
outputs="text", # Output is text
title="OCR with Donut-CORD Model", # Title of the interface
description="Upload an image to extract text using the OCR Donut-CORD model.",
)
# Launch the app
if __name__ == "__main__":
iface.launch()
|