Update Dockerfile

Dockerfile

@@ -2,7 +2,13 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install OS dependencies
+# Set HOME to a writable location
+ENV HOME=/app
+ENV MPLCONFIGDIR=/app/.config/matplotlib
+ENV TORCH_HOME=/app/.torch
+ENV STREAMLIT_HOME=/app/.streamlit
+
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -19,22 +25,22 @@ RUN apt-get update && apt-get install -y \
     zlib1g-dev \
     && rm -rf /var/lib/apt/lists/*
 
+# Create required writable folders
+RUN mkdir -p /app/.torch /app/.streamlit /app/.config/matplotlib
+
 # Copy project files
 COPY requirements.txt ./
 COPY src/ ./src/
 
-# Install pip packages
+# 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'
 
-# Expose Streamlit port
 EXPOSE 8501
 
-# Health check endpoint
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-# Start Streamlit app
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]