luck210 commited on
Commit
a9dd3d9
·
verified ·
1 Parent(s): fc96b33

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +17 -0
Dockerfile CHANGED
@@ -1,6 +1,23 @@
1
  FROM python:3.9-slim
 
 
2
  RUN pip install fastapi uvicorn pydantic requests transformers torch python-multipart
 
 
 
 
 
3
  COPY app.py /app/app.py
 
 
4
  WORKDIR /app
 
 
 
 
 
 
5
  EXPOSE 7860
 
 
6
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
1
  FROM python:3.9-slim
2
+
3
+ # Installer les dépendances
4
  RUN pip install fastapi uvicorn pydantic requests transformers torch python-multipart
5
+
6
+ # Créer un répertoire de cache accessible
7
+ RUN mkdir -p /app/cache && chmod 777 /app/cache
8
+
9
+ # Copier le code
10
  COPY app.py /app/app.py
11
+
12
+ # Définir le dossier de travail
13
  WORKDIR /app
14
+
15
+ # Définir le cache pour transformers
16
+ ENV TRANSFORMERS_CACHE=/app/cache
17
+ ENV HF_HOME=/app/cache
18
+
19
+ # Exposer le port pour Hugging Face Spaces
20
  EXPOSE 7860
21
+
22
+ # Démarrer FastAPI
23
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]