Update Dockerfile

Dockerfile

@@ -7,21 +7,33 @@ RUN apt-get update && apt-get install -y git git-lfs
 # Initialize git-lfs
 RUN git-lfs install
 
-# Set the working directory
-WORKDIR /app
+# Create a non-root user and set up environment variables
+RUN useradd -m -u 9000 user
+
+# Set environment variables
+ENV HOME=/home/user \
+    TRANSFORMERS_CACHE=/home/user/.cache \
+    PATH=/home/user/.local/bin:$PATH
+
+# Switch to the newly created user
+USER user
+
+# Set the working directory inside the container
+WORKDIR $HOME/app
+
+# Copy the requirements file
+COPY --chown=user:user requirements.txt .
 
 # Clone the Hugging Face model repository (replace with the desired model URL)
-RUN git clone https://huggingface.co/matsant01/STEMerald-2b /app/model
+RUN git clone https://huggingface.co/matsant01/STEMerald-2b $HOME/app/model
 
-# Install necessary Python libraries
-COPY requirements.txt .
+# Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the application code
-COPY . .
+# Copy the application code and set file ownership
+COPY --chown=user:user . .
 
-# Expose the app port (if using FastAPI or another web server)
+# Expose the port and run the app with Uvicorn 
 EXPOSE 8000
-
-# Command to run the FastAPI server
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+