imgtotxt / app.py
jaimin's picture
Create app.py
93ffbec
raw
history blame
No virus
869 Bytes
import easyocr as ocr #OCR
from PIL import Image #Image Processing
import numpy as np #Image Processing
import gradio as gr
input = gr.inputs.Image(label="Upload your Image", type = type=['png','jpg','jpeg'], optional=True)
def load_model():
reader = ocr.Reader(['en'],model_storage_directory='.')
return reader
def text(image):
reader = load_model() #load model
input_image = Image.open(image) #read image
result = reader.readtext(np.array(input_image))
result_text = [] #empty list for results
for text in result:
result_text.append(text[1])
return result_text
output = gr.outputs.Textbox(type="text",label="Captions")
title = "Image Captioning "
interface = gr.Interface(
fn=text,
inputs = input,
theme="grass",
outputs=output,
title=title,
)
interface.launch(debug=True)