Update ocr_engine.py
Browse files- ocr_engine.py +6 -3
ocr_engine.py
CHANGED
|
@@ -1,8 +1,11 @@
|
|
| 1 |
import easyocr
|
|
|
|
| 2 |
|
| 3 |
-
|
|
|
|
|
|
|
| 4 |
|
| 5 |
def extract_text(image):
|
|
|
|
| 6 |
results = reader.readtext(image)
|
| 7 |
-
|
| 8 |
-
return text
|
|
|
|
| 1 |
import easyocr
|
| 2 |
+
import streamlit as st
|
| 3 |
|
| 4 |
+
@st.cache_resource
|
| 5 |
+
def load_ocr():
|
| 6 |
+
return easyocr.Reader(['en'], gpu=False)
|
| 7 |
|
| 8 |
def extract_text(image):
|
| 9 |
+
reader = load_ocr()
|
| 10 |
results = reader.readtext(image)
|
| 11 |
+
return " | ".join([r[1] for r in results])
|
|
|