Set hugging face cache locations

Dockerfile

@@ -1,8 +1,7 @@
 FROM python:3.9
 
 WORKDIR /code
-
-ENV HF_HOME=$HOME/code/huggingface
+RUN mkdir -p /code && chmod -R 777 /code
 
 COPY ./requirements.txt /code/requirements.txt
 

main.py

@@ -17,6 +17,11 @@ import torch
 # torch.mps.empty_cache()
 # torch.set_num_threads(1)
 
+import os
+os.environ["HF_HOME"] = "/code/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/code/huggingface/transformers"
+
+
 app = FastAPI()
 
 class Item(BaseModel):