Update Dockerfile

Dockerfile

@@ -1,29 +1,29 @@
-# Use the official Python 3.8 image
-FROM python:3.9
-
-# set the working directory to /code
-WORKDIR /code
-
-# copy the current directory contents in the container /code
-
-COPY ./requirements.txt /code/requirements.txt
-
-# install the requirements.txt with parameter np cache
-RUN pip install --no-cache-dir --upgrade -r  /code/requirements.txt
-
-#Switching up a new user: 'user'
-USER user
-
-# set home to the user's home directory
-
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin/:$PATH
-
-# set working directory to the user's home directory
-WORKDIR $HOME/app
-
-# copy the current directory contents into the container at $HOME/app setting the owner to the user
-COPY --chown=user . $HOME/app 
-
-# start the FastAPI App on port 7860
-CMD [ "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860" ]
+# Use the official Python 3.9 image
+FROM python:3.9
+
+# Set the working directory to /code
+WORKDIR /code
+
+# Copy requirements.txt into the container
+COPY ./requirements.txt /code/requirements.txt
+
+# Install dependencies
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Switch to a new user: 'user'
+RUN useradd -m user
+USER user
+
+# Set home and cache directories for the 'user'
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin/:$PATH \
+    TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+
+# Set working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy current directory into the container and set ownership to 'user'
+COPY --chown=user . $HOME/app
+
+# Start the FastAPI app
+CMD [ "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860" ]