Update Dockerfile

Dockerfile

@@ -11,8 +11,14 @@ RUN apt-get update && apt-get install -y \
     libxext6 \
     && rm -rf /var/lib/apt/lists/*
 
-# Install PyTorch and TorchVision
-RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
+
+# Install dependencies including torch first
+RUN pip install --no-cache-dir torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
+
+# Then, install other requirements
+COPY --mount=target=/tmp/requirements.txt,source=requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
+
 
 # Install Detectron2 from source
-RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
+RUN pip install 'git+https://github.com/facebookresearch/detectron2.git