Update utils.py
Browse files
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
|