File size: 1,100 Bytes
810c411 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
# Image TensorFlow GPU (fonctionne aussi en CPU-only)
FROM tensorflow/tensorflow:2.15.0-gpu
# Installation des dépendances système supplémentaires
RUN apt-get update && apt-get install -y \
libgl1-mesa-glx \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
# Création de l'utilisateur user (requis par Hugging Face)
RUN useradd -m -u 1000 user
ENV HOME=/home/user
ENV PATH=/home/user/.local/bin:$PATH
# Définir le répertoire de travail
WORKDIR $HOME/app
# Copier les fichiers de requirements avec les bonnes permissions
COPY --chown=user:user requirements.txt .
# Passer à l'utilisateur user
USER user
# Installer les dépendances Python
RUN pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir -r requirements.txt
# Copier le reste de l'application
COPY --chown=user:user . .
# Port requis par Hugging Face Spaces
EXPOSE 7860
# Variables d'environnement pour Hugging Face Spaces
ENV PORT=7860
ENV HOST=0.0.0.0
# Commande de démarrage compatible HF Spaces
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860", "--log-level", "info"] |