Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official PyTorch image with CUDA support as the base image
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
 
 # Install Git and system libraries required for OpenGL without interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
@@ -40,11 +40,11 @@ WORKDIR $HOME/app
 RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
 
 # Install specific versions of PyTorch and TorchVision
-RUN pip install torchvision==0.15.2
+RUN pip install torchvision==0.10.0
 
 # Install dependencies
 #COPY requirements.txt $HOME/app/requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt gradio opencv-python scipy
+RUN pip install --no-cache-dir -r requirements.txt gradio==3.7.0 opencv-python scipy
 
 # Download checkpoint files using aria2
 RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/video-retalking/resolve/main/30_net_gen.pth -d $HOME/app/checkpoints -o 30_net_gen.pth