Update Dockerfile

Dockerfile

@@ -20,6 +20,10 @@ 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 detectron2 manually
 RUN pip3 install 'git+https://github.com/facebookresearch/detectron2.git'