Fixed cache path

Dockerfile

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

main.py

@@ -17,9 +17,9 @@ import torch
 # torch.mps.empty_cache()
 # torch.set_num_threads(1)
 
-import os
-os.environ["HF_HOME"] = "./.cache"
-os.environ["TRANSFORMERS_CACHE"] = "./.cache"
+# import os
+# os.environ["HF_HOME"] = "/.cache"
+# os.environ["TRANSFORMERS_CACHE"] = "/.cache"
 
 
 app = FastAPI()