Spaces:
Running
on
Zero
Running
on
Zero
feiyang-cai
commited on
Update utils.py
Browse files
utils.py
CHANGED
@@ -238,7 +238,7 @@ class MolecularPropertyPredictionModel():
|
|
238 |
# # handle error
|
239 |
# return "error"
|
240 |
|
241 |
-
@spaces.GPU(duration=
|
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=
|
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
|