Update utils.py

utils.py

@@ -161,6 +161,7 @@ class MolecularPropertyPredictionModel():
             finetuning_task="classification", # this is not about our task type
             trust_remote_code=True,
             auth_token = os.environ.get("TOKEN")
+        )
 
         self.base_model = AutoModelForSequenceClassification.from_pretrained(
             "ChemFM/ChemFM-3B",