Update Dockerfile

Dockerfile

@@ -23,16 +23,11 @@ RUN pip install --upgrade pip
 # Install PyTorch CPU version
 RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu --index-url https://download.pytorch.org/whl/cpu
 
-# Install other Python dependencies
+# Install Python dependencies
 RUN pip install -r requirements.txt
 
-# Clone Detectron2 repo
-RUN git clone https://github.com/facebookresearch/detectron2.git /detectron2
-
-# Checkout a commit known to work with PointRend
-RUN cd /detectron2 && git checkout 6d9d66b
-
-# Install Detectron2 in editable mode
+# Clone Detectron2 using the v0.6 tag (compatible with PointRend)
+RUN git clone --branch v0.6 https://github.com/facebookresearch/detectron2.git /detectron2
 RUN pip install -e /detectron2
 
 # Add Detectron2 to PYTHONPATH