Adding changes to requirements.txt and Dockerfile

Dockerfile

@@ -1,27 +1,59 @@
 FROM python:3.10
 
-WORKDIR /app
+# Set up user to avoid permission issues
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
-# Install system dependencies
+WORKDIR $HOME/app
+
+# Switch to root to install system packages
+USER root
+
+# Install ALL necessary system dependencies for OpenCV and GUI libraries
 RUN apt-get update && apt-get install -y \
     git \
     build-essential \
+    cmake \
+    python3-dev \
+    # OpenCV dependencies
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
+    libglib2.0-0 \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    libgtk2.0-dev \
+    # Additional useful libraries
+    wget \
+    unzip \
+    ffmpeg \
+    libopencv-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Install torch CPU version first
-RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu torchaudio==2.0.2+cpu \
-    --extra-index-url https://download.pytorch.org/whl/cpu
+# Switch back to user
+USER user
 
-# Clone and install detectron2
-RUN git clone https://github.com/facebookresearch/detectron2 && \
-    pip install -e detectron2
+# Copy requirements first (for better caching)
+COPY --chown=user:user requirements.txt .
 
-# Copy requirements and install
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+# Install Python dependencies
+RUN pip install --user --no-cache-dir --upgrade pip && \
+    pip install --user --no-cache-dir torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cpu && \
+    pip install --user --no-cache-dir -r requirements.txt
+
+# Install detectron2 separately after torch is installed
+RUN git clone https://github.com/facebookresearch/detectron2 $HOME/app/detectron2 && \
+    pip install --user -e $HOME/app/detectron2
 
 # Copy app files
-COPY . .
+COPY --chown=user:user . .
+
+# Expose port for Gradio
+EXPOSE 7860
 
-# Run the app
+# Run the application
 CMD ["python", "app.py"]

requirements.txt

@@ -1,37 +1,22 @@
+# Don't include torch/torchvision/torchaudio here since they're in Dockerfile
+# Don't include detectron2 since it's installed separately in Dockerfile
 
-# Core dependencies - PyTorch first
-
-# CRITICAL: Use pre-built detectron2 wheel for torch 2.0 CPU
-
-# Rest of your dependencies
-numpy>=1.23.0
-Pillow>=8.3.2
-opencv-python
-git+https://github.com/cocodataset/panopticapi.git
-git+https://github.com/mcordts/cityscapesScripts.git
-imutils
-cython
-setuptools==59.5.0
+# Core dependencies
+gradio==4.19.2
+numpy
+opencv-python-headless==4.8.1.78  # Use headless version for Docker
+pillow
+matplotlib
 scipy
-shapely
-h5py
-submitit
-scikit-image
-timm==0.4.12
+timm
 einops
-icecream
-wandb
-ftfy
-regex
-inflect
-diffdist
-gdown
-huggingface_hub
-pytorch_lightning
-tqdm
-gradio
-wget
 
-# NATTEN should be installed last
--f https://shi-labs.com/natten/wheels/cpu/torch2.0.0/index.html
-natten==0.14.6
+# natten with correct torch version
+natten==0.14.6+torch200cpu -f https://shi-labs.com/natten/wheels/cpu/torch2.0.0/index.html
+
+# Additional dependencies
+diffdist
+pytest
+ftfy
+iopath
+pycocotools