Update Dockerfile

Dockerfile

@@ -2,7 +2,6 @@ FROM ubuntu:latest
 
 WORKDIR /content
 
-# Install essential system packages
 RUN apt-get update && apt-get install -y \
     python3 \
     python3-pip \
@@ -24,93 +23,56 @@ RUN apt-get update && apt-get install -y \
     libgtk2.0-0 \
     jq \
     libraw1394-dev \
-    libopenblas-dev \
-    ffmpeg
+    libopenblas-dev 
 
-# Alias python3 to python
 RUN alias python=python3    
 
-# Add universe repository
+RUN apt-get install -y gnupg wget htop sudo git git-lfs software-properties-common build-essential cmake curl 
+RUN apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavdevice-dev libgl1 libgtk2.0-0 jq libraw1394-dev libopenblas-dev
 RUN apt-add-repository -y universe
 
-# Create and activate Python virtual environment
 RUN python3 -m venv /content/venv
 
-# Install required Python packages
-RUN /content/venv/bin/pip install \
-    git+https://github.com/One-sixth/fairseq.git \
-    setuptools \
-    pandas \
-    scipy \
-    matplotlib \
-    torch \
-    torchvision \
-    ffmpeg-python \
-    imageio[ffmpeg] \
-    tensorboardX \
-    huggingface-hub \
-    g2p_en \
-    opencv-python \
-    imageio \
-    torchaudio \
-    gradio \
-    gtts \
-    soundfile \
-    numpy==2.2.0 \
-    mediapipe \
-    gfpgan \
-    cmake==3.24.1.1 \
-    altair \
-    imageio-ffmpeg \
-    pocketsphinx \
-    jq \
-    nltk
-
-# Clone and install PyVideoFramesExtractor
-RUN git clone https://github.com/chi0tzp/PyVideoFramesExtractor && \
-    cd PyVideoFramesExtractor && \
-    /content/venv/bin/pip install -r requirements.txt
+RUN /content/venv/bin/pip install git+https://github.com/One-sixth/fairseq.git
+RUN /content/venv/bin/pip install setuptools pandas scipy matplotlib torch torchvision ffmpeg-python imageio[ffmpeg] tensorboardX huggingface-hub g2p_en opencv-python imageio torchaudio gradio gtts soundfile fairseq huggingface-hub g2p_en altair imageio-ffmpeg pocketsphinx ffmpeg jq 
+# RUN /content/venv/bin/pip install setuptools pandas scipy matplotlib torch torchvision ffmpeg-python imageio[ffmpeg] tensorboardX huggingface-hub g2p_en opencv-python imageio torchaudio gradio gtts soundfile fairseq huggingface-hub g2p_en altair imageio-ffmpeg pocketsphinx ffmpeg jq nltk
 
-# Install git-lfs and clone additional repositories
-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 /content/venv/bin/pip install numpy==2.2.0
+RUN /content/venv/bin/pip install cmake==3.24.1.1 
+RUN /content/venv/bin/pip install gfpgan
+RUN /content/venv/bin/pip install mediapipe
+# RUN git clone https://github.com/TencentARC/GFPGAN.git && \
+#     cd GFPGAN && \
+#     /content/venv/bin/pip install basicsr && \
+#     /content/venv/bin/pip install facexlib && \
+#     /content/venv/bin/pip install -r requirements.txt && \
+#     python3 setup.py develop && \
+#     /content/venv/bin/pip install realesrgan
 
-RUN git clone https://huggingface.co/camenduru/one-shot-talking-face-20.04-t4 one-shot-talking-face && \
-    cd one-shot-talking-face && \
-    /content/venv/bin/pip install -r requirements.txt && \
-    chmod 755 OpenFace/FeatureExtraction
+RUN git clone https://github.com/chi0tzp/PyVideoFramesExtractor && cd PyVideoFramesExtractor && /content/venv/bin/pip install -r requirements.txt
 
-# Update one-shot-talking-face scripts to use CPU
+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 && /content/venv/bin/pip install -r requirements.txt && chmod 755 OpenFace/FeatureExtraction
 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
 RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
-
-# Prepare output directory
 RUN mkdir /content/out
 
-# Copy application code
 COPY app.py /content/app.py
 
-# Set up admin user
+
 RUN adduser --disabled-password --gecos '' admin
 RUN adduser admin sudo
 RUN echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers
 
-# Set permissions
 RUN chown -R admin:admin /content
 RUN chmod -R 777 /content
 RUN chown -R admin:admin /home
 RUN chmod -R 777 /home
-
-# Switch to admin user
 USER admin
 
-# Expose application port
 EXPOSE 7860
 
-# Run application
 CMD ["/content/venv/bin/python", "app.py"]