From 3b52c2ba023c210ffb9c0ea56d6edb5592b34b7d Mon Sep 17 00:00:00 2001 From: MangoPig Date: Sat, 21 Feb 2026 23:07:54 +0000 Subject: [PATCH] Download separately --- Code/docker-compose.yaml | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/Code/docker-compose.yaml b/Code/docker-compose.yaml index c426229..25d7a87 100644 --- a/Code/docker-compose.yaml +++ b/Code/docker-compose.yaml @@ -18,10 +18,15 @@ services: command: - | # Download Chinese OCR models if missing (first run only) - if [ ! -f /opt/app-root/src/.cache/docling/models/EasyOcr/zh_sim_g2.pth ] || [ ! -f /opt/app-root/src/.cache/docling/models/EasyOcr/zh_tra_g2.pth ]; then - echo "Downloading Chinese OCR models (Simplified + Traditional)..." - python3 -c "import easyocr; easyocr.Reader(['en','ch_sim','ch_tra'], gpu=False)" - # Copy downloaded models to correct location + # Note: ch_sim and ch_tra are mutually exclusive, must download separately + if [ ! -f /opt/app-root/src/.cache/docling/models/EasyOcr/zh_sim_g2.pth ]; then + echo "Downloading Simplified Chinese OCR model..." + python3 -c "import easyocr; easyocr.Reader(['en','ch_sim'], gpu=False)" + cp -n /opt/app-root/src/.EasyOCR/model/*.pth /opt/app-root/src/.cache/docling/models/EasyOcr/ 2>/dev/null || true + fi + if [ ! -f /opt/app-root/src/.cache/docling/models/EasyOcr/zh_tra_g2.pth ]; then + echo "Downloading Traditional Chinese OCR model..." + python3 -c "import easyocr; easyocr.Reader(['en','ch_tra'], gpu=False)" cp -n /opt/app-root/src/.EasyOCR/model/*.pth /opt/app-root/src/.cache/docling/models/EasyOcr/ 2>/dev/null || true fi exec uvicorn docling_serve.app:app --host 0.0.0.0 --port 5001