Update Dockerfile

Dockerfile

@@ -1,21 +1,18 @@
 # Use an official Python runtime as a parent image
 FROM python:3.9
 
-# Install CUDA for GPU support (optional, based on your needs)
-# Uncomment the following lines if you need GPU support
-# RUN apt-get update && apt-get install -y --no-install-recommends \
-#     cuda-cudart-dev-11.0 \
-#   && \
-#   rm -rf /var/lib/apt/lists/*
-
 # Set the working directory in the container
 WORKDIR /app
 
+# Create a group and user for running the application
+RUN groupadd mygroup && useradd -ms /bin/bash -G mygroup appuser
+
 # Copy the current directory contents into the container at /app
-COPY . /app
+# Use --chown to set the ownership and --chmod to set the permissions
+COPY --chown=appuser:mygroup . /app
 
 # Install any needed packages specified in requirements.txt
-RUN pip install  -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Make port 7860 available to the world outside this container
 EXPOSE 7860
@@ -24,10 +21,13 @@ EXPOSE 7860
 ENV GRADIO_SERVER_PORT=7860
 
 # Set the HF_HOME environment variable to a writable directory
-ENV HF_HOME=/app/.cache/huggingface/hub
+ENV HF_HOME=/app/.cache/huggingface
+
+# Create the cache directory if it doesn't exist and set its ownership
+RUN mkdir -p $HF_HOME && chown -R appuser:mygroup $HF_HOME
 
-# Create the cache directory if it doesn't exist
-RUN mkdir -p $HF_HOME
+# Switch to the appuser to run the application
+USER appuser
 
 # Run app.py when the container launches
 CMD ["python", "app.py"]