Upload processor/redaction.py with huggingface_hub

processor/redaction.py

@@ -1,9 +1,15 @@
 import cv2
-import easyocr
 import numpy as np
 
-# Initialize the EasyOCR reader (loads model on first run)
-reader = easyocr.Reader(['en'], gpu=False)  # Set gpu=True if applicable
+# Lazy-loaded EasyOCR reader (only initialized when redact_names is called)
+_reader = None
+
+def _get_reader():
+    global _reader
+    if _reader is None:
+        import easyocr
+        _reader = easyocr.Reader(['en'], gpu=False)
+    return _reader
 
 def redact_names(img: np.ndarray) -> np.ndarray:
     """
@@ -16,7 +22,7 @@ def redact_names(img: np.ndarray) -> np.ndarray:
     h, w = img.shape[:2]
     
     # Run OCR on the image
-    results = reader.readtext(img)
+    results = _get_reader().readtext(img)
     
     has_text = False
     for (bbox, text, prob) in results: