Mise à jour du Dockerfile et des dépendances pour Hugging Face Spaces

Dockerfile

@@ -2,34 +2,25 @@ FROM python:3.11-slim
 
 WORKDIR /app
 
-# 1. Mise à jour de pip seule
-RUN pip install --upgrade pip
-
-# 2. Installation des dépendances système
+# Installation des dépendances système
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# 3. Installation des dépendances Python
+# Copie des requirements d'abord pour le cache Docker
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 4. Installation de PyTorch séparément avec timeout augmenté
-RUN pip install --no-cache-dir --timeout=600 \
-    torch==2.0.1+cpu \
-    torchvision==0.15.2+cpu \
-    -f https://download.pytorch.org/whl/torch_stable.html
-
-# 5. Copie du code
+# Copie du code source
 COPY . .
 
-# Configuration Streamlit
+# Configuration des variables d'environnement
 ENV STREAMLIT_SERVER_PORT=8501
 ENV STREAMLIT_SERVER_HEADLESS=true
 ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
 
-# Port exposé
-EXPOSE 8501
+# Se déplacer dans le dossier src
+WORKDIR /app/src
 
 # Commande de démarrage
 CMD ["streamlit", "run", "streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]

requirements.txt

@@ -1,6 +1,10 @@
-streamlit>=1.28.0
-transformers>=4.53.0
-torch>=2.0.0
-pillow>=10.0.0
-huggingface-hub>=0.19.0
-accelerate>=0.25.0
+streamlit==1.28.0
+transformers==4.36.2
+torch==2.0.1
+torchvision==0.15.2
+pillow==10.0.0
+huggingface-hub==0.19.4
+accelerate==0.25.0
+fastapi==0.104.1
+uvicorn==0.24.0
+python-multipart==0.0.6