fffiloni commited on
Commit
c27f5f3
·
verified ·
1 Parent(s): d62e849

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +40 -3
Dockerfile CHANGED
@@ -5,7 +5,21 @@ ENV DEBIAN_FRONTEND=noninteractive
5
  # Set the MKL_THREADING_LAYER environment variable to GNU
6
  ENV MKL_THREADING_LAYER=GNU
7
 
8
- RUN apt-get update && apt-get install -y git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential cmake
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
  RUN useradd -m -u 1000 user
11
 
@@ -25,15 +39,38 @@ ENV HOME=/home/user \
25
  # Set the working directory to the user's home directory
26
  WORKDIR $HOME/app
27
 
 
 
 
 
28
  RUN git clone -b dev https://github.com/fffiloni/dreamtalk $HOME/app
29
 
 
30
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O $HOME/app/checkpoints/denoising_network.pth
31
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
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
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
  RUN pip install gradio
36
 
 
 
 
37
  COPY app.py .
38
 
39
  # Set the environment variable to specify the GPU device
 
5
  # Set the MKL_THREADING_LAYER environment variable to GNU
6
  ENV MKL_THREADING_LAYER=GNU
7
 
8
+ # Install system dependencies including those required for dlib
9
+ RUN apt-get update && apt-get install -y \
10
+ git \
11
+ wget \
12
+ libgl1-mesa-glx \
13
+ libglib2.0-0 \
14
+ ffmpeg \
15
+ libx264-dev \
16
+ build-essential \
17
+ cmake \
18
+ libopenblas-dev \
19
+ liblapack-dev \
20
+ libx11-dev \
21
+ libgtk-3-dev \
22
+ python3-dev
23
 
24
  RUN useradd -m -u 1000 user
25
 
 
39
  # Set the working directory to the user's home directory
40
  WORKDIR $HOME/app
41
 
42
+ # Print detailed Python information
43
+ RUN python -c "import sys; print(f'Python {sys.version}')"
44
+
45
+ # Clone the repository
46
  RUN git clone -b dev https://github.com/fffiloni/dreamtalk $HOME/app
47
 
48
+ # Download model checkpoints
49
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O $HOME/app/checkpoints/denoising_network.pth
50
  RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
51
 
52
+ # Install Python dependencies
53
+ RUN pip install --no-cache-dir \
54
+ urllib3==1.26.6 \
55
+ transformers==4.28.1 \
56
+ yacs==0.1.8 \
57
+ scipy==1.10.1 \
58
+ scikit-image==0.20.0 \
59
+ scikit-learn==1.2.2 \
60
+ PyYAML==6.0 \
61
+ Pillow==9.5.0 \
62
+ numpy==1.24.2 \
63
+ opencv-python==4.7.0.72 \
64
+ imageio==2.27.0 \
65
+ ffmpeg-python \
66
+ av==11.0.0 \
67
+ moviepy==1.0.3
68
+
69
  RUN pip install gradio
70
 
71
+ # Install dlib with verbose output
72
+ RUN pip install --verbose --no-cache-dir dlib
73
+
74
  COPY app.py .
75
 
76
  # Set the environment variable to specify the GPU device