Rakib023 commited on
Commit
5a1b6d0
·
verified ·
1 Parent(s): 4bfe574

Update ocr_engine.py

Browse files
Files changed (1) hide show
  1. ocr_engine.py +6 -3
ocr_engine.py CHANGED
@@ -1,8 +1,11 @@
1
  import easyocr
 
2
 
3
- reader = easyocr.Reader(['en'])
 
 
4
 
5
  def extract_text(image):
 
6
  results = reader.readtext(image)
7
- text = " | ".join([res[1] for res in results])
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])