Image-Processing-Pipeline / models /text_extraction_model.py
todap's picture
Upload 5 files
9cfa91c verified
raw
history blame contribute delete
267 Bytes
import easyocr
class TextExtractionModel:
def __init__(self):
self.reader = easyocr.Reader(['en'])
def extract_text(self, image_path):
result = self.reader.readtext(image_path)
return ' '.join([detection[1] for detection in result])