Update Dockerfile

Dockerfile

@@ -1,11 +1,5 @@
 # syntax=docker/dockerfile:1
 
-# Comments are provided throughout this file to help you get started.
-# If you need more help, visit the Dockerfile reference guide at
-# https://docs.docker.com/go/dockerfile-reference/
-
-# Want to help us make this template better? Share your feedback here: https://forms.gle/ybq9Krt8jtBL3iCk7
-
 ARG PYTHON_VERSION=3.12.3
 FROM python:${PYTHON_VERSION}-slim as base
 
@@ -16,10 +10,10 @@ ENV PYTHONDONTWRITEBYTECODE=1
 # the application crashes without emitting any logs due to buffering.
 ENV PYTHONUNBUFFERED=1
 
+# Set the working directory
 WORKDIR /app
 
 # Create a non-privileged user that the app will run under.
-# See https://docs.docker.com/go/dockerfile-user-best-practices/
 ARG UID=10001
 RUN adduser \
     --disabled-password \
@@ -30,38 +24,31 @@ RUN adduser \
     --uid "${UID}" \
     appuser
 
-# Download dependencies as a separate step to take advantage of Docker's caching.
-# Leverage a cache mount to /root/.cache/pip to speed up subsequent builds.
-# Leverage a bind mount to requirements.txt to avoid having to copy them into
-# into this layer.
-RUN --mount=type=cache,target=/root/.cache/pip \
-    --mount=type=bind,source=requirements.txt,target=requirements.txt \
-    python -m pip install -r requirements.txt
+# Copy requirements.txt before other files to leverage Docker's caching mechanism
+COPY requirements.txt .
 
+# Install dependencies
+RUN --mount=type=cache,target=/root/.cache/pip \
+    python -m pip install --no-cache-dir -r requirements.txt
 
-# Create a directory named 'data' and assign its ownership to appuser
-RUN mkdir -p /data
-RUN chown appuser /data
+# Create necessary directories with appropriate permissions
+RUN mkdir -p /data /app/.cache/huggingface && \
+    chown -R appuser:appuser /data /app/.cache/huggingface && \
+    chmod -R 777 /data /app/.cache/huggingface
 
+# Set the TRANSFORMERS_CACHE and HF_HOME environment variables
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+ENV HF_HOME=/app/.cache/huggingface
 
 # Switch to the non-privileged user to run the application.
 USER appuser
 
-# Set the TRANSFORMERS_CACHE environment variable
-ENV TRANSFORMERS_CACHE=/tmp/.cache/huggingface
-
-# Create the cache folder with appropriate permissions
-RUN mkdir -p $TRANSFORMERS_CACHE && chmod -R 777 $TRANSFORMERS_CACHE
-
-
-# Copy the source code into the container.
+# Copy the rest of the source code into the container.
 COPY . .
 
-# Expose the port that the application listens on.
+# Expose the ports that the application listens on.
 EXPOSE 7860
 EXPOSE 8501
 
-# Run the application.
 # Run the application.
 CMD ["bash", "-c", "uvicorn app:app --host 0.0.0.0 --port 7860 & streamlit run frontend.py --server.port 8501 --server.enableXsrfProtection false"]
-