Update Dockerfile

Dockerfile

@@ -2,6 +2,14 @@ FROM python:3.9-slim
 
 WORKDIR /code
 
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Set up cache directory with proper permissions
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+RUN mkdir -p /home/user/.cache/huggingface && \
+    chown -R user:user /home/user/.cache
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
@@ -17,6 +25,12 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the application code
 COPY . .
 
+# Change ownership of the working directory
+RUN chown -R user:user /code
+
+# Switch to non-root user
+USER user
+
 # Expose the port the app runs on
 EXPOSE 7860