Mauricio-100 commited on
Commit
cb24b78
·
verified ·
1 Parent(s): a743dbc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +16 -11
Dockerfile CHANGED
@@ -1,36 +1,41 @@
1
- # Étape 1 : Base
2
  FROM python:3.10-slim
3
 
4
- # Étape 2 : Variables d'environnement
5
  ENV DEBIAN_FRONTEND=noninteractive
6
  ENV PYTHONUNBUFFERED=1
7
  ENV HF_TOKEN=${HF_TOKEN}
8
 
9
- # Étape 3 : Installer les dépendances système minimales
10
- RUN apt-get update && apt-get install -y git && apt-get clean
11
 
12
- # Étape 4 : Copier ton projet dans le conteneur
13
  WORKDIR /app
14
  COPY . /app
15
 
16
- # Étape 5 : Installer les dépendances Python
17
  RUN pip install --no-cache-dir --upgrade pip
18
  RUN pip install --no-cache-dir \
 
19
  transformers \
20
  datasets \
21
  accelerate \
22
  torch \
23
  flask \
24
- huggingface_hub \
25
  sentencepiece \
26
  tokenizers \
27
  tqdm
28
 
29
- # Étape 6 : Authentification Hugging Face (utilise la variable HF_TOKEN)
30
- RUN python -m huggingface_hub.login --token ${HF_TOKEN}
 
 
 
 
 
31
 
32
- # Étape 7 : Exposer le port
33
  EXPOSE 7860
34
 
35
- # Étape 8 : Lancer ton serveur Flask
36
  CMD ["python", "server/api.py"]
 
1
+ # Étape 1 : Base légère
2
  FROM python:3.10-slim
3
 
4
+ # Étape 2 : Environnement
5
  ENV DEBIAN_FRONTEND=noninteractive
6
  ENV PYTHONUNBUFFERED=1
7
  ENV HF_TOKEN=${HF_TOKEN}
8
 
9
+ # Étape 3 : Outils système
10
+ RUN apt-get update && apt-get install -y git curl && apt-get clean
11
 
12
+ # Étape 4 : Répertoire de travail
13
  WORKDIR /app
14
  COPY . /app
15
 
16
+ # Étape 5 : Dépendances Python
17
  RUN pip install --no-cache-dir --upgrade pip
18
  RUN pip install --no-cache-dir \
19
+ huggingface_hub \
20
  transformers \
21
  datasets \
22
  accelerate \
23
  torch \
24
  flask \
 
25
  sentencepiece \
26
  tokenizers \
27
  tqdm
28
 
29
+ # Étape 6 : Authentification Hugging Face (nouvelle syntaxe)
30
+ # Vérifie si la variable HF_TOKEN est définie avant de se loguer
31
+ RUN if [ -n "${HF_TOKEN}" ]; then \
32
+ huggingface-cli login --token ${HF_TOKEN}; \
33
+ else \
34
+ echo "⚠️ Aucun token Hugging Face fourni (HF_TOKEN manquant)"; \
35
+ fi
36
 
37
+ # Étape 7 : Port de l’API
38
  EXPOSE 7860
39
 
40
+ # Étape 8 : Lancement du serveur Flask
41
  CMD ["python", "server/api.py"]