Update Dockerfile.nvidia

Dockerfile.nvidia

@@ -17,6 +17,9 @@ RUN DEBIAN_FRONTEND=noninteractive apt update && apt install -y \
     unzip \
     && rm -rf /var/lib/apt/lists/*
 
+# Install ffmpeg-python
+RUN pip install ffmpeg-python
+
 # Download and extract buffalo_l model for InsightFace
 RUN wget https://github.com/deepinsight/insightface/releases/download/v0.7/buffalo_l.zip -O /tmp/buffalo_l.zip && \
     mkdir -p /root/.insightface/models/buffalo_l && \
@@ -24,16 +27,6 @@ RUN wget https://github.com/deepinsight/insightface/releases/download/v0.7/buffa
     unzip /tmp/buffalo_l.zip && \
     rm -f /tmp/buffalo_l.zip
 
-# Install TensorRT via NVIDIA repository (more reliable)
-RUN apt update && apt install -y \
-    gnupg2 \
-    curl \
-    lsb-release \
-    && curl -sSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/nvidia-cuda-toolkit_11.8.0-1_amd64.deb -o nvidia-cuda-toolkit_11.8.0-1_amd64.deb \
-    && dpkg -i nvidia-cuda-toolkit_11.8.0-1_amd64.deb \
-    && apt-get install -f -y \
-    && rm nvidia-cuda-toolkit_11.8.0-1_amd64.deb
-
 # Install the Refacer repository and dependencies
 RUN git clone https://github.com/xaviviro/refacer && \
     cd refacer && \