File size: 2,500 Bytes
207d493
fe30080
 
207d493
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
010ab61
 
5d84704
 
 
fe30080
5d54c3f
fe30080
4b7c058
fe30080
010ab61
5d54c3f
 
d467ddf
 
 
 
 
125c3bf
 
d467ddf
 
 
 
 
 
 
 
 
 
 
 
 
 
fe30080
4a01eab
54f17e5
bfd6141
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
FROM ubuntu:latest

WORKDIR /content

RUN apt-get update && apt-get install -y \
    python3 \
    python3-pip \
    gnupg \
    wget \
    htop \
    sudo \
    git \
    git-lfs \
    software-properties-common \
    build-essential \
    cmake \
    curl \
    libavcodec-dev \
    libavformat-dev \
    libavdevice-dev \
    libgl1 \
    libgtk2.0-0 \
    jq \
    libraw1394-dev \
    libopenblas-base
    
RUN alias python=python3    

RUN apt-get install -y gnupg wget htop sudo git git-lfs software-properties-common build-essential cmake curl 
RUN apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavdevice-dev libgl1 libgtk2.0-0 jq libraw1394-dev libopenblas-base
RUN apt-add-repository -y universe

RUN pip3 install pandas scipy matplotlib torch torchvision ffmpeg-python imageio[ffmpeg] tensorboardX huggingface-hub g2p_en opencv-python fairseq imageio torchaudio gradio gtts soundfile fairseq huggingface-hub g2p_en altair imageio-ffmpeg pocketsphinx dlib ffmpeg jq "numpy==1.23.1"

RUN pip install cmake==3.24.1.1

RUN git clone https://github.com/TencentARC/GFPGAN.git && cd GFPGAN && pip install basicsr && pip install facexlib && pip install -r requirements.txt && python3 setup.py develop && pip install realesrgan
RUN git clone https://github.com/chi0tzp/PyVideoFramesExtractor && cd PyVideoFramesExtractor && pip install -r requirements.txt

RUN git lfs install
RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphinx && cd pocketsphinx && cmake -S . -B build && cmake --build build --target install
RUN git clone https://huggingface.co/camenduru/one-shot-talking-face-20.04-t4 one-shot-talking-face && cd one-shot-talking-face && pip install -r requirements.txt && chmod 755 OpenFace/FeatureExtraction
RUN sed -i 's/.cuda()/ /' one-shot-talking-face/test_script.py
RUN sed -i 's/.cuda()/ /' one-shot-talking-face/tools/interface.py
RUN sed -i 's/.load(checkpoint_path)/.load(checkpoint_path,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
RUN mkdir /content/out

COPY app.py /content/app.py


RUN adduser --disabled-password --gecos '' admin
RUN adduser admin sudo
RUN echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers

RUN chown -R admin:admin /content
RUN chmod -R 777 /content
RUN chown -R admin:admin /home
RUN chmod -R 777 /home
USER admin

EXPOSE 7860

CMD ["python3", "app.py"]