Add token

app.py

@@ -66,9 +66,13 @@ MODEL_DETAILS = {
 class PIIMaskingModel:
     def __init__(self, model_name: str):
         self.model_name = model_name
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-        self.model = AutoModelForTokenClassification.from_pretrained(model_name)
-        # No need to specify device as we are forcing CPU usage
+        hf_token = st.secrets["HF_TOKEN"]  # Retrieve the token from secrets
+        self.tokenizer = AutoTokenizer.from_pretrained(
+            model_name, use_auth_token=hf_token
+        )
+        self.model = AutoModelForTokenClassification.from_pretrained(
+            model_name, use_auth_token=hf_token
+        )
 
     def process_text(
         self, text: str