Update Dockerfile

Dockerfile

@@ -2,6 +2,17 @@ FROM python:3.11-slim
 
 WORKDIR /code
 
+# Create cache directory and set permissions
+RUN mkdir -p /cache && \
+    chmod 777 /cache && \
+    mkdir -p /.cache && \
+    chmod 777 /.cache
+
+# Set cache environment variables
+ENV TRANSFORMERS_CACHE=/cache \
+    HF_HOME=/cache \
+    XDG_CACHE_HOME=/cache
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     gcc \
@@ -27,6 +38,14 @@ COPY ./app.py /code/app.py
 COPY ./templates /code/templates
 COPY ./models /code/models
 
+# Set proper permissions for the application
+RUN chown -R 1000:1000 /code && \
+    chmod -R 755 /code
+
 EXPOSE 7860
 
+# Create and switch to non-root user
+RUN useradd -m -u 1000 user
+USER user
+
 CMD ["python", "app.py"]