pnicewiczoig commited on
Commit
d6f9d03
·
1 Parent(s): c34c3fc

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -0
app.py ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import easyocr as ocr #OCR
2
+ import streamlit as st #Web App
3
+ from PIL import Image #Image Processing
4
+ import numpy as np #Image Processing
5
+
6
+ import re
7
+
8
+ #title
9
+ st.title("OCR - ML Example")
10
+
11
+ #subtitle
12
+ st.markdown("## Optical Character Recognition")
13
+
14
+ #image uploader
15
+ image = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpeg'])
16
+
17
+
18
+ @st.cache
19
+ def load_model():
20
+ reader = ocr.Reader(['en'],model_storage_directory='.')
21
+ return reader
22
+
23
+ reader = load_model() #load model
24
+
25
+ if image is not None:
26
+
27
+ input_image = Image.open(image) #read image
28
+ st.image(input_image) #display image
29
+
30
+ with st.spinner("Please wait ... processing"):
31
+
32
+
33
+ result = reader.readtext(np.array(input_image))
34
+
35
+ result_text = [] #empty list for results
36
+
37
+
38
+ for text in result:
39
+ # only output if numbers and spaces
40
+ if re.match("^[0-9 ]+$", text[1]):
41
+ result_text.append(text[1])
42
+
43
+ st.write(result_text)
44
+ st.balloons()
45
+ else:
46
+ st.write("Upload an Image")