Update Dockerfile

Dockerfile

@@ -1,6 +1,8 @@
 # Use Python 3.9 slim as base image
 FROM python:3.9-slim
 
+# Create a non-root user
+RUN useradd -m -u 1000 user
 
 # Set working directory
 WORKDIR /app
@@ -14,6 +16,7 @@ RUN apt-get update && apt-get install -y \
     libxrender-dev \
     libgomp1 \
     wget \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy requirements first for better caching
@@ -22,21 +25,27 @@ COPY requirements.txt .
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-
-
 # Copy application files
 COPY . .
 
-# Create directory for model cache
-RUN mkdir -p /root/.cache/carvekit
+# Create cache directories with proper permissions
+RUN mkdir -p /app/.cache/carvekit && \
+    mkdir -p /home/user/.cache/carvekit && \
+    chown -R user:user /app && \
+    chown -R user:user /home/user
 
-# Expose the port
-EXPOSE 7860
+# Switch to non-root user
+USER user
 
 # Set environment variables
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT=7860
 ENV PYTHONUNBUFFERED=1
+ENV HOME=/home/user
+ENV CARVEKIT_CACHE_DIR=/app/.cache/carvekit
+
+# Expose the port
+EXPOSE 7860
 
 # Health check
 HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \