Set transformers_cache variable to not use root folder

Dockerfile

@@ -12,6 +12,7 @@ RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
 COPY src/baseline.py src/baseline.py
+ENV TRANSFORMERS_CACHE=/coma-fixer/.cache
 RUN python src/baseline.py  # This pre-downloads models and tokenizers
 
 COPY . .
@@ -27,9 +28,10 @@ FROM python:3.10-slim as deploy
 WORKDIR /comma-fixer
 COPY --from=base /comma-fixer /comma-fixer
 COPY --from=base /venv /venv
-# Copy pre-downloaded models and make sure we are using the env
-COPY --from=base /root/.cache/huggingface/hub/ /root/.cache/huggingface/hub/
 ENV PATH="/venv/bin:$PATH"
+# Copy pre-downloaded models and make sure we are using the env
+ENV TRANSFORMERS_CACHE=/coma-fixer/.cache
+COPY --from=base /coma-fixer/.cache /coma-fixer/.cache
 
 EXPOSE 8000
 CMD uvicorn "app:app" --port 8000 --host "0.0.0.0"