feiyang-cai commited on
Commit
e039bb2
·
verified ·
1 Parent(s): a3c849b

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -233,7 +233,7 @@ class MolecularPropertyPredictionModel():
233
  # handle error
234
  return "error"
235
 
236
- @spaces.GPU
237
  def predict(self, valid_df, task_type):
238
  test_dataset = Dataset.from_pandas(valid_df)
239
  # construct the dataloader
 
233
  # handle error
234
  return "error"
235
 
236
+ @spaces.GPU(duration=20)
237
  def predict(self, valid_df, task_type):
238
  test_dataset = Dataset.from_pandas(valid_df)
239
  # construct the dataloader