Spaces:

CordwainerSmith
/

GolemPII

Running on CPU Upgrade

cordwainersmith commited on 21 days ago

Commit

5584918

•

1 Parent(s): 5d89fbb

Add token

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ MODEL_DETAILS = {
 class PIIMaskingModel:
     def __init__(self, model_name: str):
         self.model_name = model_name
-        hf_token = st.secrets["HF_TOKEN"]  # Retrieve the token from secrets
         self.tokenizer = AutoTokenizer.from_pretrained(
             model_name, use_auth_token=hf_token
         )

 class PIIMaskingModel:
     def __init__(self, model_name: str):
         self.model_name = model_name
+        hf_token = st.secrets["hf_token"]  # Retrieve the token from secrets
         self.tokenizer = AutoTokenizer.from_pretrained(
             model_name, use_auth_token=hf_token
         )