Update Dockerfile

Dockerfile

@@ -1,22 +1,31 @@
 FROM python:3.9-slim
 
+# Create a non-root user
+RUN useradd -m appuser
+
+# Set environment variables for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
+
 # Set working directory
 WORKDIR /app
 
-# Install dependencies
+# Copy requirements and install them
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Set writable cache directory
-ENV TRANSFORMERS_CACHE=/app/cache
-ENV HF_HOME=/app/cache
-RUN mkdir -p /app/cache
+# Create cache directory and set permissions
+RUN mkdir -p /app/cache && chown -R appuser /app/cache
 
-# Copy all files
+# Copy all files and set ownership
 COPY . .
+RUN chown -R appuser /app
+
+# Switch to non-root user
+USER appuser
 
 # Expose port for Streamlit
 EXPOSE 7860
 
-# Run the app
+# Run the Streamlit app
 CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]