fffiloni commited on
Commit
9232754
1 Parent(s): 30dfaa7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -2,7 +2,7 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
5
- RUN apt-get update && apt-get install -y git wget
6
 
7
  RUN useradd -m -u 1000 user
8
 
@@ -27,7 +27,8 @@ RUN git clone -b dev https://github.com/camenduru/dreamtalk $HOME/app
27
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O $HOME/app/checkpoints/denoising_network.pth
28
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
29
 
30
- RUN pip install -q yacs av gradio
 
31
 
32
  COPY app.py .
33
 
 
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
5
+ RUN apt-get update && apt-get install -y git wget ffmpeg
6
 
7
  RUN useradd -m -u 1000 user
8
 
 
27
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O $HOME/app/checkpoints/denoising_network.pth
28
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
29
 
30
+ # Install dependencies
31
+ RUN pip install --no-cache-dir -r requirements.txt gradio urllib==1.26.6 transformers==4.28.1 dlib
32
 
33
  COPY app.py .
34