Update utils.py

utils.py

@@ -204,9 +204,9 @@ class MolecularPropertyPredictionModel():
         # switch adapter
         try:
             self.adapter_name = adapter_name
-            self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, auth_token = os.environ.get("TOKEN_FROM_SECRET") or True)
+            self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
             if adapter_name not in self.apapter_scaler_path:
-                self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", auth_token = os.environ.get("TOKEN_FROM_SECRET") or True)
+                self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
             if os.path.exists(self.apapter_scaler_path[adapter_name]):
                 self.scaler = pickle.load(open(self.apapter_scaler_path[adapter_name], "rb"))
             else: