Update utils.py

utils.py

@@ -160,22 +160,21 @@ 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_FROM_SECRET") or True
-        )
+            auth_token = os.environ.get("TOKEN")
 
         self.base_model = AutoModelForSequenceClassification.from_pretrained(
             "ChemFM/ChemFM-3B",
             config=config,
             device_map="cpu",
             trust_remote_code=True,
-            auth_token = os.environ.get("TOKEN_FROM_SECRET") or True
+            auth_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_FROM_SECRET") or True
+            auth_token = os.environ.get("TOKEN")
         )
         special_tokens_dict = dict(pad_token=DEFAULT_PAD_TOKEN)
         smart_tokenizer_and_embedding_resize(