update

app.py

@@ -3,6 +3,7 @@ from huggingface_hub import HfApi, get_collection, list_collections
 from utils import MolecularPropertyPredictionModel, dataset_task_types, dataset_descriptions, dataset_property_names, dataset_property_names_to_dataset
 import pandas as pd
 import os
+import spaces
 
 def get_models():
     # this is the collection id for the molecular property prediction models
@@ -26,6 +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)
 def predict_single_label(smiles, property_name):
     property_id = dataset_property_names_to_dataset[property_name]
 

utils.py

@@ -266,7 +266,7 @@ class MolecularPropertyPredictionModel():
                 # handle error
                 return "error"
     
-    @spaces.GPU(duration=60)
+
     def predict(self, valid_df, task_type):
 
         with calculateDuration("predicting"):