feiyang-cai commited on
Commit
62cd725
·
verified ·
1 Parent(s): b1e4cef

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -238,7 +238,7 @@ class MolecularPropertyPredictionModel():
238
  # # handle error
239
  # return "error"
240
 
241
- @spaces.GPU(duration=5)
242
  def predict(self, valid_df, task_type):
243
  test_dataset = Dataset.from_pandas(valid_df)
244
  # construct the dataloader
@@ -266,7 +266,7 @@ class MolecularPropertyPredictionModel():
266
 
267
  return y_pred
268
 
269
- @spaces.GPU(duration=30)
270
  def predict_long(self, valid_df, task_type):
271
  test_dataset = Dataset.from_pandas(valid_df)
272
  # construct the dataloader
 
238
  # # handle error
239
  # return "error"
240
 
241
+ @spaces.GPU(duration=10)
242
  def predict(self, valid_df, task_type):
243
  test_dataset = Dataset.from_pandas(valid_df)
244
  # construct the dataloader
 
266
 
267
  return y_pred
268
 
269
+ @spaces.GPU(duration=60)
270
  def predict_long(self, valid_df, task_type):
271
  test_dataset = Dataset.from_pandas(valid_df)
272
  # construct the dataloader