feiyang-cai commited on
Commit
dd57ba6
·
1 Parent(s): 5c66cc6
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -28,6 +28,7 @@ import spaces
28
  #property_names = list(candidate_models.keys())
29
  model = MolecularGenerationModel()
30
 
 
31
  def predict_single_label(logp, tpas, sas, qed, logp_choose, tpsa_choose, sas_choose, qed_choose):
32
  input_dict = dict()
33
  if logp_choose:
 
28
  #property_names = list(candidate_models.keys())
29
  model = MolecularGenerationModel()
30
 
31
+ @spaces.GPU(duration=60)
32
  def predict_single_label(logp, tpas, sas, qed, logp_choose, tpsa_choose, sas_choose, qed_choose):
33
  input_dict = dict()
34
  if logp_choose: