Pretrained cache directory

main.py

@@ -7,14 +7,15 @@ import os
 
 app = FastAPI()
 os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
 os.makedirs(os.environ["HF_HOME"], exist_ok=True)
 
 model_path = hf_hub_download(repo_id="MienOlle/sentiment_analysis_api",
                              filename="sentimentAnalysis.pth",
                              cache_dir=os.environ["HF_HOME"]
                              )
-modelToken = token.from_pretrained("mdhugol/indonesia-bert-sentiment-classification")
-model = modelSC.from_pretrained("mdhugol/indonesia-bert-sentiment-classification", num_labels=3)
+modelToken = token.from_pretrained("mdhugol/indonesia-bert-sentiment-classification", cache_dir=os.environ["TRANSFORMERS_CACHE"])
+model = modelSC.from_pretrained("mdhugol/indonesia-bert-sentiment-classification", num_labels=3, cache_dir=os.environ["TRANSFORMERS_CACHE"])
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.load_state_dict(torch.load(model_path, map_location=torch.device(device)))