EasyOCR-model / model.py
anjali2002's picture
Initial commit of EasyOCR model
b4959be
raw
history blame contribute delete
645 Bytes
import easyocr
from typing import List
class EasyOCRModel:
def __init__(self):
self.reader = easyocr.Reader(['en']) # Initialize with English; add languages if needed.
def predict(self, image_path: str) -> List[str]:
"""
Perform OCR on the given image.
Args:
image_path (str): Path to the input image.
Returns:
List[str]: Extracted text from the image.
"""
return self.reader.readtext(image_path, detail=0)
# Test the model locally
if __name__ == "__main__":
model = EasyOCRModel()
result = model.predict("sample_image.jpg")
print(result)