Update Dockerfile

Dockerfile

@@ -31,19 +31,32 @@ RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install --no-cache-dir llama-cpp-python==
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Create cache directory with proper permissions
+RUN mkdir -p /app/.cache/huggingface
+RUN chown -R user:user /app
+
 # Copy app code
 COPY . .
+RUN chown -R user:user /app
 
 # Create models directory
-RUN mkdir -p models
+RUN mkdir -p models && chown -R user:user models
 
-# Expose Gradio port
-EXPOSE 7860
+# Switch to non-root user
+USER user
 
 # Environment variables
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT=7860
 ENV CUDA_VISIBLE_DEVICES=0
+ENV HF_HOME=/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+
+# Expose Gradio port
+EXPOSE 7860
 
 # Start app
 CMD ["python", "app.py"]