Update Dockerfile

Dockerfile

@@ -3,7 +3,7 @@ FROM ubuntu:latest
 WORKDIR /content
 
 RUN apt-get update && apt-get install -y \
-    python3.8 \
+    python3 \
     python3-pip \
     gnupg \
     wget \
@@ -31,8 +31,8 @@ RUN apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavdevice-dev lib
 RUN apt-add-repository -y universe
 
 RUN pip install git+https://github.com/One-sixth/fairseq.git --break-system-packages
-RUN pip3 install distutils "setuptools==44.0.0" --break-system-packages
-RUN pip3 install "numpy==1.22.0" --break-system-packages
+# RUN pip3 install distutils "setuptools==44.0.0" --break-system-packages
+RUN pip3 install numpy --break-system-packages
 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 --break-system-packages
 
 RUN pip install cmake==3.24.1.1 --break-system-packages
@@ -44,6 +44,7 @@ RUN git clone https://github.com/chi0tzp/PyVideoFramesExtractor && cd PyVideoFra
 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 --break-system-packages && chmod 755 OpenFace/FeatureExtraction
+COPY util.py /content/one-shot-talking-face/models/util.py
 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