import easyocr as ocr #OCR | |
import streamlit as st #Web App | |
from PIL import Image #Image Processing | |
import numpy as np #Image Processing | |
#title | |
st.title("Easy OCR - Extract Text from Images") | |
#subtitle | |
st.markdown("## Using `easyocr`, `streamlit`") | |
#image uploader | |
image = st.file_uploader(label = "Upload your image here", type=['png', 'jpg', 'jpeg']) | |
# @st.cache | |
# def load_model(): | |
# reader = ocr.Reader(['en']) | |
# return reader | |
# | |
# reader = load_model() #load model | |
if image is not None: | |
input_image = Image.open(image) #read image | |
st.image(input_image) #display image | |
with st.spinner("🤖 AI is at Work! "): | |
reader = ocr.Reader(['en']) | |
result = reader.readtext(np.array(input_image)) | |
result_text = [] | |
for text in result: | |
result_text.append(text[1]) | |
st.write(result_text) | |
st.success("Here you go!") | |
else: | |
st.write("Upload an Image") | |
st.caption("Made with ❤️ by z") | |