Commit ·
c331a85
1
Parent(s): d98f65b
Update Dockerfile
Browse files- 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 |
-
|
| 19 |
-
|
| 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
|