Translate-Image / app.py
Omnibus's picture
Update app.py
8f6282a
raw
history blame
929 Bytes
import gradio as gr
import pandas as pd
import easyocr
import torch
import PIL
from PIL import Image
from PIL import ImageDraw
def draw_boxes(image, bounds, color='blue', width=1):
draw = ImageDraw.Draw(image)
for bound in bounds:
p0, p1, p2, p3 = bound[0]
draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
return image
def detect(img, lang=['en']):
reader = easyocr.Reader(lang)
bounds = reader.readtext(img)
im = PIL.Image.open(img)
im_out=draw_boxes(im, bounds)
return im_out,pd.DataFrame(bounds).iloc[: 0, -1:],pd.DataFrame(bounds).iloc[: 0, -1:]
with gr.Blocks() as robot:
im=gr.Image(type="filepath")
go_btn=gr.Button()
out_im=gr.Image()
with gr.Row():
out_txt=gr.Textbox(lines=8)
data_f=gr.Dataframe(headers=['text', 'confidence'])
go_btn.click(detect,im,[out_im,out_txt,data_f])
robot.queue(concurrency_count=10).launch()