File size: 869 Bytes
93ffbec |
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 27 28 29 30 31 32 33 |
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) |