Update Dockerfile

Dockerfile

@@ -6,14 +6,20 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Create a user with explicit UID 1000 (common for first non-root user)
+RUN useradd -m -u 1000 appuser
+
 # Set working directory
 WORKDIR /app
 
-# Create a directory for Triton cache with proper permissions
-RUN mkdir -p /app/.triton && chmod 777 /app/.triton
+# Create directories for various caches with proper permissions
+RUN mkdir -p /app/.triton /app/.torch_cache && \
+    chown -R appuser:appuser /app
 
-# Set environment variable to use the custom Triton cache directory
+# Set environment variables for cache directories
 ENV TRITON_CACHE_DIR=/app/.triton
+ENV TORCH_HOME=/app/.torch_cache
+ENV TORCHINDUCTOR_CACHE_DIR=/app/.torch_cache/inductor
 
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
@@ -24,6 +30,12 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the rest of the application
 COPY . .
 
+# Ensure proper ownership of all files
+RUN chown -R appuser:appuser /app
+
+# Switch to the appuser
+USER appuser
+
 # Expose the port Streamlit runs on
 EXPOSE 8501