Set .cache

Dockerfile

@@ -2,6 +2,7 @@ FROM python:3.9
 
 WORKDIR /code
 RUN mkdir -p /code && chmod -R 777 /code
+RUN mkdir -p ./.cache && chmod -R 777 ./.cache
 
 COPY ./requirements.txt /code/requirements.txt
 

main.py

@@ -18,8 +18,8 @@ import torch
 # torch.set_num_threads(1)
 
 import os
-os.environ["HF_HOME"] = "/code/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/code/huggingface/transformers"
+os.environ["HF_HOME"] = "./.cache"
+os.environ["TRANSFORMERS_CACHE"] = "./.cache"
 
 
 app = FastAPI()