Spaces:
Sleeping
Sleeping
import easyocr | |
import gradio as gr | |
from PIL import Image | |
reader = easyocr.Reader(['en']) | |
def inference(img_path, width_ths): | |
output = reader.readtext(img_path, detail=0, slope_ths=0.7, ycenter_ths=0.9, | |
height_ths=0.8, width_ths=width_ths, add_margin=0.2) | |
output = "\n".join(output) | |
title = "Receipt RAG" | |
description = "A simple Gradio interface to query receipts using RAG" | |
examples = [["data/receipt_00000.JPG", 7.7], | |
["data/receipr_00001.jpg", 7.7]] | |
demo = gr.Interface(inference, | |
inputs = [gr.Image(width=320, height=320, label="Input Receipt"), | |
gr.Slider(0, 10, 7.7, 0.1, label="Width Threshold to merge bounding boxes")], | |
outputs= [gr.Textbox(label="OCR Output", type="text")], | |
title=title, | |
description=description, | |
examples=examples) | |
demo.launch() | |