updating reqs and docker

Dockerfile

@@ -1,5 +1,6 @@
 FROM python:3.9-slim
 
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     gcc \
@@ -10,14 +11,25 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
+# Set Matplotlib temp folder
 ENV MPLCONFIGDIR=/tmp/matplotlib
 
-COPY requirements.txt /app/requirements.txt
+# Set working directory
 WORKDIR /app
 
+# Copy requirements first and install them
+COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
-COPY . /app
+# Install torch separately (before detectron2)
+RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu torchaudio==2.0.2+cpu --index-url https://download.pytorch.org/whl/cpu
 
+# Install detectron2 separately
+RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
+
+# Copy full app code
+COPY . .
+
+# Run
 CMD ["python3", "app.py"]

requirements.txt

@@ -9,7 +9,3 @@ labelme2coco
 numpy
 from_root
 gradio==3.50.2
-torch==2.0.1+cpu
-torchvision==0.15.2+cpu
-torchaudio==2.0.2+cpu
-git+https://github.com/facebookresearch/detectron2.git