Update Dockerfile

Dockerfile

@@ -1,16 +1,29 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
-
+# Use the official Python image
 FROM python:3.11
 
+# Set the working directory inside the container
 WORKDIR /code
 
+# Copy the requirements file and install dependencies
 COPY ./requirements.txt /code/requirements.txt
-
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN mkdir -p /.cache/huggingface/hub
+# Create the Hugging Face cache directory, set permissions, and create a non-root user
+RUN mkdir -p /.cache/huggingface/hub && \
+    chmod -R 777 /.cache/huggingface && \
+    useradd -m nonrootuser
+
+# Set ownership of the .cache directory to the non-root user
+RUN chown -R nonrootuser:nonrootuser /.cache/huggingface
 
+# Copy the rest of the application code
 COPY . .
 
+# Change ownership of the application code to the non-root user
+RUN chown -R nonrootuser:nonrootuser /code
+
+# Switch to the non-root user
+USER nonrootuser
+
+# Specify the command to run the application
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]