rag-service/ocr-service/app/models.py

17 lines
338 B
Python

import os
from .engine import PaddleOcrEngine
def load_runtime_engine() -> PaddleOcrEngine | None:
if os.getenv("OCR_LOAD_ENGINE") != "1":
return None
try:
return PaddleOcrEngine()
except Exception:
return None
if __name__ == "__main__":
PaddleOcrEngine()
print("PaddleOCR models loaded")