Update Dockerfile

Dockerfile

@@ -3,15 +3,15 @@ FROM python:3.12-slim
 # Set up a new user with user ID 1000
 RUN useradd -m -u 1000 user
 
-# Switch to the "user" user
-USER user
+# Switch to root user temporarily for installing system dependencies
+USER root
 
 # Set environment variables for the user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
 # Set the working directory to the user's home directory
-WORKDIR $HOME/app
+WORKDIR /home/user/app
 
 # Upgrade pip
 RUN pip install --no-cache-dir --upgrade pip==23.3.1
@@ -21,6 +21,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential && \
     rm -rf /var/lib/apt/lists/*
 
+# Switch back to the "user" user
+USER user
+
 # Copy the application code into the container at $HOME/app and set the ownership to the "user"
 COPY --chown=user product_return_prediction $HOME/app/product_return_prediction
 COPY --chown=user README.md $HOME/app/