Update app/main.py

app/main.py

@@ -13,8 +13,14 @@ import numpy as np
 
 app = FastAPI()
 
-model = BertForSequenceClassification.from_pretrained("JaySenpai/bert-model")
-tokenizer = BertTokenizer.from_pretrained("JaySenpai/bert-model")
+model = BertForSequenceClassification.from_pretrained(
+    "JaySenpai/bert-model",
+    cache_dir="/code/cache"
+)
+tokenizer = BertTokenizer.from_pretrained(
+    "JaySenpai/bert-model",
+    cache_dir="/code/cache"
+)
 model.eval()
 
 le = LabelEncoder()