Spaces:
Running
Running
Update Dockerfile
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|