Update main.py

main.py

@@ -1,15 +1,18 @@
+
+import os
+
+# Set safe Hugging Face cache directory before any transformers import
+os.environ["HF_HOME"] = "/tmp/hf-cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf-cache"
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-import os
-
-# Set cache directory to avoid permission errors
-os.environ["HF_HOME"] = "/tmp/hf-cache"
 
 app = FastAPI()
 
-# Load tokenizer and model once at startup
+# Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("cybersectony/phishing-email-detection-distilbert_v2.4.1")
 model = AutoModelForSequenceClassification.from_pretrained("cybersectony/phishing-email-detection-distilbert_v2.4.1")