MemeScope / utils /ocr_utils.py
vishal1232006's picture
Update utils/ocr_utils.py
fc802e2 verified
Raw History Blame Contribute Delete
405 Bytes
import easyocr
from PIL import Image
import numpy as np
# Initialize EasyOCR reader (English for now)
reader = easyocr.Reader(['en'])
def extract_text(image):
"""
Extract text from an image using EasyOCR.
"""
if isinstance(image, Image.Image):
image = np.array(image)
results = reader.readtext(image)
text = " ".join([res[1] for res in results])
return text.strip()