Marcin-XStudio commited on
Commit
df5399a
·
1 Parent(s): a3a70b9
Files changed (1) hide show
  1. Dockerfile +7 -2
Dockerfile CHANGED
@@ -3,8 +3,9 @@ FROM python:3.10-slim
3
  # 1) Variables HF avant tout
4
  ENV HF_HOME="/home/user/.cache/huggingface" \
5
  HF_HUB_CACHE="/home/user/.cache/huggingface/hub" \
6
- TRANSFORMERS_CACHE="/home/user/.cache/huggingface/transformers"
7
- # ENV DOCLING_ARTIFACTS_PATH="/home/user/.cache/docling/models"
 
8
 
9
 
10
  # 2) Créer l’utilisateur non-root
@@ -32,6 +33,10 @@ RUN mkdir -p /home/user/app/model_cache $HF_HUB_CACHE \
32
 
33
  # 6) Télécharger le modèle au build-time sous l’utilisateur non-root
34
  USER user
 
 
 
 
35
  RUN python - <<EOF
36
  import os
37
  from huggingface_hub import snapshot_download
 
3
  # 1) Variables HF avant tout
4
  ENV HF_HOME="/home/user/.cache/huggingface" \
5
  HF_HUB_CACHE="/home/user/.cache/huggingface/hub" \
6
+ TRANSFORMERS_CACHE="/home/user/.cache/huggingface/transformers" \
7
+ DOCLING_ARTIFACTS_PATH="/home/user/.cache/docling/models" \
8
+ HOME="/home/user"
9
 
10
 
11
  # 2) Créer l’utilisateur non-root
 
33
 
34
  # 6) Télécharger le modèle au build-time sous l’utilisateur non-root
35
  USER user
36
+
37
+ RUN pip install --no-cache-dir docling \
38
+ && docling-tools models download
39
+
40
  RUN python - <<EOF
41
  import os
42
  from huggingface_hub import snapshot_download