Spaces:
Sleeping
Sleeping
File size: 874 Bytes
06ca3b1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | #!/usr/bin/env python3
"""Docker build: load OCR models only (no Orchestrator / no LLM). Used by Dockerfile.worker.ocr."""
from __future__ import annotations
import logging
import os
import sys
_APP_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if _APP_ROOT not in sys.path:
sys.path.insert(0, _APP_ROOT)
os.chdir(_APP_ROOT)
from dotenv import load_dotenv
load_dotenv()
from app.runtime_env import apply_runtime_env_defaults
apply_runtime_env_defaults()
logging.basicConfig(level=logging.INFO, format="%(levelname)s %(name)s | %(message)s")
logger = logging.getLogger("prewarm_ocr_worker")
def main() -> None:
from vision_ocr.pipeline import OcrVisionPipeline
logger.info("Loading OcrVisionPipeline...")
OcrVisionPipeline()
logger.info("OCR worker prewarm finished successfully.")
if __name__ == "__main__":
main()
|