Update Dockerfile

Dockerfile

@@ -14,6 +14,11 @@ 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()/ /' test_script.py
+RUN sed -i 's/.cuda()/ /' tools/interface.py
+RUN sed -i 's/.load(checkpoint_path)/.load(checkpoint_path,map_location='cpu') /' tools/interface.py
+RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location='cpu') /' tools/interface.py
+
 RUN mkdir /content/out
 
 COPY app.py /content/app.py