JimSmith007 commited on
Commit
941d7ce
·
1 Parent(s): ded8479

Fix installation albumentations pour Hugging Face

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -11,9 +11,10 @@ WORKDIR /home/user/app
11
  # Installe les dépendances
12
  COPY --chown=user requirements.txt requirements.txt
13
  RUN pip install --no-cache-dir -r requirements.txt
 
14
 
15
  # Copie tous les fichiers du projet
16
  COPY --chown=user . .
17
 
18
  # Lance l'app FastAPI sur le port 7860
19
- CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
11
  # Installe les dépendances
12
  COPY --chown=user requirements.txt requirements.txt
13
  RUN pip install --no-cache-dir -r requirements.txt
14
+ RUN pip install --no-cache-dir albumentations[imgaug] albumentations[opencv] albumentations[pytorch]
15
 
16
  # Copie tous les fichiers du projet
17
  COPY --chown=user . .
18
 
19
  # Lance l'app FastAPI sur le port 7860
20
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]