Update utils.py

utils.py

@@ -160,7 +160,7 @@ class MolecularPropertyPredictionModel():
             num_labels=1,
             finetuning_task="classification", # this is not about our task type
             trust_remote_code=True,
-            auth_token = os.environ.get("TOKEN")
+            token = os.environ.get("TOKEN")
         )
 
         self.base_model = AutoModelForSequenceClassification.from_pretrained(
@@ -168,14 +168,14 @@ class MolecularPropertyPredictionModel():
             config=config,
             device_map="cpu",
             trust_remote_code=True,
-            auth_token = os.environ.get("TOKEN")
+            token = os.environ.get("TOKEN")
         )
 
         # load the tokenizer
         self.tokenizer = AutoTokenizer.from_pretrained(
             "ChemFM/admet_ppbr_az",
             trust_remote_code=True,
-            auth_token = os.environ.get("TOKEN")
+            token = os.environ.get("TOKEN")
         )
         special_tokens_dict = dict(pad_token=DEFAULT_PAD_TOKEN)
         smart_tokenizer_and_embedding_resize(