pragnakalp commited on
Commit
5d54c3f
1 Parent(s): 1e54c58

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -5
Dockerfile CHANGED
@@ -6,10 +6,13 @@ RUN apt-get install -y gnupg wget htop sudo git git-lfs software-properties-comm
6
  RUN apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavdevice-dev libgl1 libgtk2.0-0 jq libdc1394-22-dev libraw1394-dev libopenblas-base
7
 
8
 
9
- RUN pip3 install pandas scipy matplotlib torch tensorboardX opencv-python imageio torchvision torchaudio gradio gtts soundfile fairseq huggingface-hub g2p_en altair imageio-ffmpeg pocketsphinx dlib ffmpeg jq "numpy==1.23.1"
10
 
11
  RUN pip install cmake==3.24.1.1
12
 
 
 
 
13
  RUN git lfs install
14
  RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphinx && cd pocketsphinx && cmake -S . -B build && cmake --build build --target install
15
  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
@@ -17,14 +20,10 @@ RUN sed -i 's/.cuda()/ /' one-shot-talking-face/test_script.py
17
  RUN sed -i 's/.cuda()/ /' one-shot-talking-face/tools/interface.py
18
  RUN sed -i 's/.load(checkpoint_path)/.load(checkpoint_path,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
19
  RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
20
- RUN cat one-shot-talking-face/tools/interface.py
21
  RUN mkdir /content/out
22
 
23
  COPY app.py /content/app.py
24
 
25
- RUN git clone https://github.com/TencentARC/GFPGAN.git && cd GFPGAN && pip install basicsr && pip install facexlib && pip install -r requirements.txt && python setup.py develop && pip install realesrgan
26
-
27
- RUN git clone https://github.com/chi0tzp/PyVideoFramesExtractor && cd PyVideoFramesExtractor && pip install -r requirements.txt
28
 
29
  RUN adduser --disabled-password --gecos '' admin
30
  RUN adduser admin sudo
 
6
  RUN apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavdevice-dev libgl1 libgtk2.0-0 jq libdc1394-22-dev libraw1394-dev libopenblas-base
7
 
8
 
9
+ 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"
10
 
11
  RUN pip install cmake==3.24.1.1
12
 
13
+ RUN git clone https://github.com/TencentARC/GFPGAN.git && cd GFPGAN && pip install basicsr && pip install facexlib && pip install -r requirements.txt && python setup.py develop && pip install realesrgan
14
+ RUN git clone https://github.com/chi0tzp/PyVideoFramesExtractor && cd PyVideoFramesExtractor && pip install -r requirements.txt
15
+
16
  RUN git lfs install
17
  RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphinx && cd pocketsphinx && cmake -S . -B build && cmake --build build --target install
18
  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
 
20
  RUN sed -i 's/.cuda()/ /' one-shot-talking-face/tools/interface.py
21
  RUN sed -i 's/.load(checkpoint_path)/.load(checkpoint_path,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
22
  RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
 
23
  RUN mkdir /content/out
24
 
25
  COPY app.py /content/app.py
26
 
 
 
 
27
 
28
  RUN adduser --disabled-password --gecos '' admin
29
  RUN adduser admin sudo