Update Dockerfile

Dockerfile

@@ -23,13 +23,16 @@ 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 Python dependencies
+# Install other Python dependencies
 RUN pip install -r requirements.txt
 
-# Clone Detectron2 (with PointRend configs)
+# Clone Detectron2 repo
 RUN git clone https://github.com/facebookresearch/detectron2.git /detectron2
 
-# Install core Detectron2
+# Checkout a commit known to work with PointRend
+RUN cd /detectron2 && git checkout 6d9d66b
+
+# Install Detectron2 in editable mode
 RUN pip install -e /detectron2
 
 # Add Detectron2 to PYTHONPATH
@@ -38,5 +41,5 @@ ENV PYTHONPATH="/detectron2:${PYTHONPATH}"
 # Expose FastAPI port
 EXPOSE 7860
 
-# Start FastAPI app (points to app.py:app)
+# Start FastAPI app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]