update

app.py

@@ -27,7 +27,7 @@ def get_description(property_name):
     property_id = dataset_property_names_to_dataset[property_name]
     return dataset_descriptions[property_id]
 
-@spaces.GPU(duration=20)
+@spaces.GPU(duration=10)
 def predict_single_label(smiles, property_name):
     property_id = dataset_property_names_to_dataset[property_name]
 
@@ -65,6 +65,7 @@ def predict_single_label(smiles, property_name):
 
     return prediction, "Prediction is done"
 
+@spaces.GPU(duration=30)
 def predict_file(file, property_name):
     property_id = dataset_property_names_to_dataset[property_name]
     try:

utils.py

@@ -230,7 +230,6 @@ class MolecularPropertyPredictionModel():
                 assert dataset_task_types[adapter_name] == "classification", f"{adapter_name} is not a regression task."
 
         self.base_model.to("cuda")
-        print(self.base_model.device)
     
     def swith_adapter(self, adapter_name, adapter_id):
         # return flag:
@@ -248,7 +247,6 @@ class MolecularPropertyPredictionModel():
                 #self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
                 #self.lora_model.to("cuda")
                 #print(self.lora_model)
-                print(self.base_model.device, "before")
                 self.base_model.set_adapter(adapter_name)
                 self.base_model.eval()
 
@@ -260,7 +258,6 @@ class MolecularPropertyPredictionModel():
                     self.scaler = None
 
                 self.adapter_name = adapter_name
-                print(self.base_model.device, "after")
                 return "switched"
             except Exception as e:
                 # handle error