13 lines
479 B
Python
13 lines
479 B
Python
from docling.document_converter import DocumentConverter
|
|
|
|
print("Warming up docling models...")
|
|
|
|
try:
|
|
# Use the sample document to warm up docling
|
|
test_file = "/app/warmup_ocr.pdf"
|
|
print(f"Using {test_file} to warm up docling...")
|
|
DocumentConverter().convert(test_file)
|
|
print("Docling models warmed up successfully")
|
|
except Exception as e:
|
|
print(f"Docling warm-up completed with: {e}")
|
|
# This is expected - we just want to trigger the model downloads
|