updating reqs and docker

Dockerfile

@@ -1,11 +1,8 @@
 FROM python:3.9-slim
 
-# --- Install system dependencies ---
 RUN apt-get update && apt-get install -y \
     git \
     gcc \
-    g++ \
-    build-essential \
     libglib2.0-0 \
     libsm6 \
     libxext6 \
@@ -13,23 +10,14 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
-# --- 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
+COPY requirements.txt /app/requirements.txt
+WORKDIR /app
 
-# --- Install basic Python dependencies ---
-COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
-# --- Install Detectron2 ---
-RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
-
-# --- Copy app code ---
 COPY . /app
-WORKDIR /app
 
-# --- Run the Gradio app ---
 CMD ["python3", "app.py"]

requirements.txt

@@ -1,4 +1,3 @@
-streamlit>=1.30
 opencv-python
 pycocotools
 Pillow
@@ -9,6 +8,8 @@ matplotlib
 labelme2coco
 numpy
 from_root
-gradio
-# torch==2.0.1
-# torchvision==0.15.2
+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