Update app.py

app.py

@@ -3,10 +3,12 @@ from pydantic import BaseModel
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
 app = FastAPI()
+token = os.getenv("HF_TOKEN")
+
 # Load the model and tokenizer
 model = AutoModelForSequenceClassification.from_pretrained(
-    "kmcs-casulit/hr_cate")
-tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/department-classification")
+    "kmcs-casulit/hr_cate", use_auth_token=token)
+tokenizer = AutoTokenizer.from_pretrained("kmcs-casulit/department-classification", use_auth_token=token)
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
 
 # Define a request model