Update app/main.py

Fix: Set writable HF cache dir for model download

app/main.py

@@ -1,3 +1,9 @@
+import os
+
+# Set a writable cache directory inside the Space container
+os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
+
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import BertTokenizer, BertForSequenceClassification
@@ -12,7 +18,7 @@ tokenizer = BertTokenizer.from_pretrained("JaySenpai/bert-model")
 model.eval()
 
 le = LabelEncoder()
-le.classes_ = np.load(r"C:\Users\jayes\Downloads\label_classes.npy", allow_pickle=True)
+le.classes_ = np.load("app/label_classes.npy", allow_pickle=True)
 
 class TextInput(BaseModel):
     text: str