26 lines
822 B
Python
26 lines
822 B
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
repo_root = os.path.dirname(__file__)
|
|
src_path = os.path.join(repo_root, "src")
|
|
if src_path not in sys.path:
|
|
sys.path.insert(0, src_path)
|
|
|
|
from utils.document_processing import create_document_converter
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
logger.info("Warming up docling models")
|
|
|
|
try:
|
|
# Use the sample document to warm up docling
|
|
test_file = "/app/warmup_ocr.pdf"
|
|
logger.info(f"Using test file to warm up docling: {test_file}")
|
|
converter = create_document_converter()
|
|
converter.convert(test_file)
|
|
logger.info("Docling models warmed up successfully")
|
|
except Exception as e:
|
|
logger.info(f"Docling warm-up completed with exception: {str(e)}")
|
|
# This is expected - we just want to trigger the model downloads
|