fffiloni commited on
Commit
5068f92
·
verified ·
1 Parent(s): 40d7ee4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
@@ -32,6 +32,7 @@ RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/
32
 
33
  # Install dependencies
34
  RUN pip install --no-cache-dir urllib3==1.26.6 transformers==4.28.1 yacs==0.1.8 scipy==1.10.1 scikit-image==0.20.0 scikit-learn==1.2.2 PyYAML==6.0 Pillow==9.5.0 numpy==1.24.2 opencv-python==4.7.0.72 imageio==2.27.0 ffmpeg-python av==11.0.0 moviepy==1.0.3 gradio==3.25.0
 
35
 
36
  COPY app.py .
37
 
 
1
+ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
 
32
 
33
  # Install dependencies
34
  RUN pip install --no-cache-dir urllib3==1.26.6 transformers==4.28.1 yacs==0.1.8 scipy==1.10.1 scikit-image==0.20.0 scikit-learn==1.2.2 PyYAML==6.0 Pillow==9.5.0 numpy==1.24.2 opencv-python==4.7.0.72 imageio==2.27.0 ffmpeg-python av==11.0.0 moviepy==1.0.3 gradio==3.25.0
35
+ RUN pip install dlib
36
 
37
  COPY app.py .
38