Update Dockerfile

Dockerfile

@@ -20,8 +20,9 @@ COPY src/ ./src/
 # Install the Python dependencies
 RUN pip3 install -r requirements.txt
 
-# Install specific versions of torch and torchvision
-RUN pip3 install torch==2.6.0+cu124 torchvision==0.21.0+cu124
+# 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 detectron2 manually