v2

app.py

@@ -58,10 +58,10 @@ class PIIMaskingModel:
         self.model_name = model_name
         hf_token = st.secrets["hf_token"]
         self.tokenizer = AutoTokenizer.from_pretrained(
-            model_name, use_auth_token=hf_token
+            model_name, token=hf_token
         )
         self.model = AutoModelForTokenClassification.from_pretrained(
-            model_name, use_auth_token=hf_token
+            model_name, token=hf_token
         )
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.model.to(self.device)
@@ -296,7 +296,7 @@ def main():
 
     if st.button("Process Text"):
         texts = [text.strip() for text in text_input.split(",") if text.strip()]
-        model = PIIMaskingModel()
+        model = PIIMaskingModel(MODEL_NAME)
 
         for text in texts:
             st.markdown(