Update utils.py

utils.py

@@ -170,7 +170,7 @@ class MolecularPropertyPredictionModel():
             trust_remote_code=True,
             token = os.environ.get("TOKEN")
         )
-        #self.base_model.to("cuda")
+        #
 
         # load the tokenizer
         self.tokenizer = AutoTokenizer.from_pretrained(
@@ -199,6 +199,7 @@ class MolecularPropertyPredictionModel():
             self.base_model.load_adapter(adapter_id, adapter_name=adapter_name, token = os.environ.get("TOKEN"))
             self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
 
+        self.base_model.to("cuda")
     
     def swith_adapter(self, adapter_name, adapter_id):
         # return flag:
@@ -232,7 +233,7 @@ class MolecularPropertyPredictionModel():
             # handle error
             return "error"
     
-    #@spaces.GPU
+    @spaces.GPU
     def predict(self, valid_df, task_type):
         test_dataset = Dataset.from_pandas(valid_df)
         # construct the dataloader