pragnakalp commited on
Commit
c331a85
1 Parent(s): d98f65b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -15,9 +15,9 @@ RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphin
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
16
  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
-
21
  RUN mkdir /content/out
22
 
23
  COPY app.py /content/app.py
 
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
16
  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