add cache to path

Dockerfile

@@ -1,6 +1,6 @@
 FROM ubuntu:22.04
 
-# Install system dependencies
+# Install necessary packages
 RUN apt-get update && apt-get install -y curl git python3 python3-pip && \
     curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash && \
     apt-get install -y git-lfs
@@ -9,18 +9,20 @@ RUN apt-get update && apt-get install -y curl git python3 python3-pip && \
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# Create a user, set environment variables, and define work directory
+# Create a non-root user and set writable directories for Matplotlib and Transformers
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
+    PATH=/home/user/.local/bin:$PATH \
+    MPLCONFIGDIR=$HOME/.config/matplotlib \
+    TRANSFORMERS_CACHE=$HOME/.cache/huggingface/hub
 
+# Prepare application directory
 WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
-# Continue with the rest of the setup
+# Setup application environment
 WORKDIR /code
-
 RUN git lfs clone https://huggingface.co/AskUI/pta-text-0.1 /code/model/
 COPY . .