Update Dockerfile

Dockerfile

@@ -20,8 +20,8 @@ COPY src/ ./src/
 # Install the Python dependencies
 RUN pip3 install -r requirements.txt
 
-# Install torch and torchvision with specific CUDA version
-RUN pip3 install torch==2.6.0+cu124 torchvision==0.21.0+cu124 -f https://download.pytorch.org/whl/cu124/torch_stable.html
+# Install torchvision 
+RUN pip3 install torchvision==0.15.2