asfafd

Dockerfile

@@ -1,56 +1,41 @@
 FROM python:3.10-slim
 
-# 1) Configure all caches under the non-root home
-ENV HOME=/home/user \
-    HF_HOME=/home/user/.cache/huggingface \
-    HF_HUB_CACHE=/home/user/.cache/huggingface/hub \
-    TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers \
-    DOCLING_ARTIFACTS_PATH=/home/user/.cache/docling/models \
-    OMP_NUM_THREADS=2
+# 1) Variables HF avant tout
+ENV HF_HOME="/home/user/.cache/huggingface" \
+    HF_HUB_CACHE="/home/user/.cache/huggingface/hub" \
+    TRANSFORMERS_CACHE="/home/user/.cache/huggingface/transformers"
 
-# 2) Install system-level dependencies
-RUN apt-get update && apt-get install -y \
-      build-essential git libpoppler-cpp-dev poppler-utils \
-      libmagic-dev python3-dev \
-    && rm -rf /var/lib/apt/lists/*
-
-# 3) Create a non-root user
+# 2) Créer l’utilisateur non-root
 RUN useradd -m -u 1000 user
 
-WORKDIR $HOME/app
+# 3) Installer dépendances système
+RUN apt-get update && apt-get install -y \
+      build-essential git libpoppler-cpp-dev poppler-utils libmagic-dev python3-dev \
+    && rm -rf /var/lib/apt/lists/*
 
-# 4) Copy & install Python runtime dependencies
-COPY requirements.txt ./
-RUN pip install --no-cache-dir \
-      torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu \
-      huggingface_hub accelerate \
-    && pip install --no-cache-dir -r requirements.txt
+# 4) Copier requirements et installer libs Python
+WORKDIR /home/user/app
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir huggingface_hub
+RUN pip install --no-cache-dir -r requirements.txt
 
-# 5) Prepare cache directories and give ownership to `user`
-RUN mkdir -p \
-      $HF_HUB_CACHE \
-      $DOCLING_ARTIFACTS_PATH \
-      $HOME/app/model_cache \
-    && chown -R user:user \
-      $HOME/.cache \
-      $HOME/app/model_cache
+# 5) Pré-créer model_cache et HF_HOME, puis chown
+RUN mkdir -p /home/user/app/model_cache $HF_HUB_CACHE \
+    && chown -R user:user /home/user/app /home/user/.cache/huggingface
 
-# 6) Switch to non-root and prefetch Docling’s models
+# 6) Télécharger le modèle au build-time sous l’utilisateur non-root
 USER user
-RUN pip install --no-cache-dir docling \
- && docling-tools models download
-
-# 7) Prefetch your NuExtract model into model_cache
-RUN python3 - <<EOF
+RUN python - <<EOF
 import os
 from huggingface_hub import snapshot_download
 snapshot_download(
-    repo_id="numind/NuExtract-1.5-tiny",
-    local_dir=os.path.join(os.getenv("HOME"), "app", "model_cache"),
-    cache_dir=os.getenv("HF_HUB_CACHE")
+  repo_id="numind/NuExtract-1.5-tiny",
+  local_dir="/home/user/app/model_cache",
+  cache_dir=os.getenv("HF_HUB_CACHE")
 )
 EOF
 
-# 8) Copy your application code & start the server
-COPY --chown=user:user . .
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# 7) Copier le reste du code et démarrer
+COPY --chown=user . .
+CMD ["uvicorn","app:app","--host","0.0.0.0","--port","7860"]