Cache directory fixes

main.py

@@ -3,11 +3,14 @@ import torch
 from transformers import AutoModelForSequenceClassification as modelSC, AutoTokenizer as token
 from fastapi import FastAPI
 from pydantic import BaseModel
+import os
 
 app = FastAPI()
+os.makedirs("/app/cache", exist_ok=True)
 
 model_path = hf_hub_download(repo_id="MienOlle/sentiment_analysis_api",
-                             filename="sentimentAnalysis.pth"
+                             filename="sentimentAnalysis.pth",
+                             cache_dir="/app/cache"
                              )
 modelToken = token.from_pretrained("mdhugol/indonesia-bert-sentiment-classification")
 model = modelSC.from_pretrained("mdhugol/indonesia-bert-sentiment-classification", num_labels=3)

requirements.txt

@@ -1,5 +1,6 @@
 fastapi 
 uvicorn 
 torch 
+pydantic 
 transformers 
 huggingface_hub