Jagrut Thakare commited on
Commit
fe5f42e
·
1 Parent(s): 2b43abc
Files changed (1) hide show
  1. Dockerfile +7 -3
Dockerfile CHANGED
@@ -25,11 +25,15 @@ WORKDIR /home/user/app
25
 
26
  # Copy files and install only Gradio
27
  COPY --chown=user:user . /home/user/app
28
- RUN pip install --user pydantic==2.8.2 gradio
 
29
 
30
- RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
31
- RUN pip install --user --no-cache-dir mxnet-cu112 onnxruntime-gpu==1.12 Cython requests==2.25.1 insightface==0.2.1 kornia==0.5.4 numpy<=2.0 dill opencv-python onnx scikit-image wandb
32
 
 
 
 
33
  COPY --chown=user:user . .
34
  RUN chmod +x download_models.sh
35
  RUN ./download_models.sh
 
25
 
26
  # Copy files and install only Gradio
27
  COPY --chown=user:user . /home/user/app
28
+ RUN pip install --user pydantic==2.8.2 gradio && \
29
+ pip cache purge
30
 
31
+ RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113 && \
32
+ pip cache purge
33
 
34
+ RUN pip install --user --no-cache-dir mxnet-cu112 onnxruntime-gpu==1.12 Cython requests==2.25.1 insightface==0.2.1 kornia==0.5.4 numpy<=2.0 dill opencv-python onnx scikit-image wandb && \
35
+ pip cache purge
36
+
37
  COPY --chown=user:user . .
38
  RUN chmod +x download_models.sh
39
  RUN ./download_models.sh