Update Dockerfile

Dockerfile

@@ -16,6 +16,14 @@ RUN apt-get update && apt-get install -y \
 WORKDIR /app
 
 COPY requirements.txt .
+
+# Install Torch for CPU first (to avoid dependency issues with detectron2)
+RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
+# Install Detectron2 with CPU flags
+RUN FORCE_CUDA=0 pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git'
+
+# Install the rest of the requirements
 RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .