Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official PyTorch image with CUDA support as the base image
-FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
 
 # Add NVIDIA CUDA GPG key
 # Add NVIDIA CUDA GPG key using a different key server
@@ -44,7 +44,7 @@ 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 torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install torch==2.0.1+cu117 torchvision==0.15.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
 
 # Install dependencies
 #COPY requirements.txt $HOME/app/requirements.txt