Update Dockerfile

Dockerfile

@@ -4,7 +4,9 @@ WORKDIR /app
 
 # Set HOME to a writable location
 ENV HOME=/app
-
+ENV TORCH_HOME=/tmp/.torch
+ENV STREAMLIT_HOME=/tmp/.streamlit
+ENV MPLCONFIGDIR=/tmp/mplconfig
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -22,26 +24,6 @@ RUN apt-get update && apt-get install -y \
     libsqlite3-dev \
     zlib1g-dev \
     && rm -rf /var/lib/apt/lists/*
-# Copy app files
-COPY app.py /app
-COPY requirements.txt /app
-
-# Install Python dependencies
-RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
-
-# Create writable directories and fix permissions
-RUN mkdir -p /app/.streamlit /app/.torch /tmp/.streamlit && \
-    chmod -R 777 /app && \
-    chmod -R 777 /tmp
-
-# Set environment variables to avoid write permission issues
-ENV TORCH_HOME=/tmp/.torch
-ENV STREAMLIT_HOME=/tmp/.streamlit
-ENV MPLCONFIGDIR=/tmp/mplconfig
-
-# Create required writable folders
-RUN mkdir -p /app/.torch /app/.streamlit /app/.config/matplotlib
 
 # Copy project files
 COPY requirements.txt ./
@@ -50,9 +32,11 @@ COPY src/ ./src/
 # Install Python dependencies
 RUN pip install --upgrade pip && \
     pip install "numpy<2.0" && \
-    pip install -r requirements.txt && \
-    pip install torchvision==0.15.2 && \
-    pip install 'git+https://github.com/facebookresearch/detectron2.git'
+    pip install -r requirements.txt
+
+# Create writable directories
+RUN mkdir -p /app/.torch /app/.streamlit /app/.config/matplotlib /tmp && \
+    chmod -R 777 /app /tmp
 
 EXPOSE 8501