requirements.txt changes

Dockerfile

@@ -1,61 +1,27 @@
-FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
-CMD nvidia-smi
+FROM python:3.10
 
-ENV DEBIAN_FRONTEND noninteractive
-RUN apt-get update && apt-get install -y \
-        git \
-        make build-essential libssl-dev zlib1g-dev \
-        libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
-        libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev  \
-    	ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
-		&& rm -rf /var/lib/apt/lists/*
-
-RUN useradd -ms /bin/bash user
-USER user
+WORKDIR /app
 
-ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
 
-RUN curl https://pyenv.run | bash
-ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
-RUN pyenv install 3.8.15 && \
-    pyenv global 3.8.15 && \
-    pyenv rehash && \
-    pip install --no-cache-dir --upgrade pip setuptools wheel
+# 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
 
-ENV WORKDIR=/code
-WORKDIR $WORKDIR
-RUN chown -R user:user $WORKDIR
-RUN chmod -R 777 $WORKDIR
+# Clone and install detectron2
+RUN git clone https://github.com/facebookresearch/detectron2 && \
+    pip install -e detectron2
 
-COPY requirements.txt $WORKDIR/requirements.txt
-RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
-RUN pip install ninja
+# Copy requirements and install
+COPY requirements.txt .
+RUN pip install -r requirements.txt
 
+# Copy app files
 COPY . .
 
-ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
-
-USER root
-RUN chown -R user:user $HOME
-RUN chmod -R 777 $HOME
-RUN chown -R user:user $WORKDIR
-RUN chmod -R 777 $WORKDIR
-
-USER user
-RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..
-RUN sh deform_setup.sh
-
-USER user
-RUN sh deform_setup.sh
-
-RUN mkdir -p examples
-RUN wget https://praeclarumjj3.github.io/files/ade20k.jpeg -P $WORKDIR/examples/
-RUN wget https://praeclarumjj3.github.io/files/cityscapes.png -P $WORKDIR/examples/
-RUN wget https://praeclarumjj3.github.io/files/coco.jpeg -P $WORKDIR/examples/
-
-USER user
-
-EXPOSE 7860
-
-ENTRYPOINT ["python", "gradio_app.py"]
+# Run the app
+CMD ["python", "app.py"]

requirements.txt

@@ -1,12 +1,7 @@
---extra-index-url https://download.pytorch.org/whl/cpu
 
 # Core dependencies - PyTorch first
-torch==2.0.1+cpu
-torchvision==0.15.2+cpu
-torchaudio==2.0.2+cpu
 
 # CRITICAL: Use pre-built detectron2 wheel for torch 2.0 CPU
-detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch2.0/index.html
 
 # Rest of your dependencies
 numpy>=1.23.0