updating docker

Dockerfile

@@ -1,9 +1,11 @@
 FROM python:3.9-slim
 
-# Install system dependencies
+# --- Install system dependencies ---
 RUN apt-get update && apt-get install -y \
     git \
     gcc \
+    g++ \
+    build-essential \
     libglib2.0-0 \
     libsm6 \
     libxext6 \
@@ -11,21 +13,23 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
-# Install PyTorch CPU version
+# --- Set environment variables ---
+ENV MPLCONFIGDIR=/tmp/matplotlib
+
+# --- Install PyTorch (CPU version) ---
 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 required Python packages
-RUN pip install opencv-python Pillow gdown gradio==4.25.0 pycocotools scikit-image shapely
+# --- Install basic Python dependencies ---
+COPY requirements.txt .
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
 
-# Install detectron2 manually
+# --- Install Detectron2 ---
 RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
 
-# Set Matplotlib temp folder
-ENV MPLCONFIGDIR=/tmp/matplotlib
-
-# Copy code
+# --- Copy app code ---
 COPY . /app
 WORKDIR /app
 
-# Run Gradio app
+# --- Run the Gradio app ---
 CMD ["python3", "app.py"]

requirements.txt

@@ -10,3 +10,5 @@ labelme2coco
 numpy
 from_root
 gradio
+# torch==2.0.1
+# torchvision==0.15.2