Spaces:

keras-io
/

molecular-property-prediction

Runtime error

vumichien commited on Jun 12, 2022

Commit

1e81845

1 Parent(s): 4c86b48

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -136,7 +136,7 @@ def graphs_from_smiles(smiles_list):
         tf.ragged.constant(pair_indices_list, dtype=tf.int64),
     )
-model = from_pretrained_keras("keras-io/wgan-molecular-graphs")
 def predict(smiles, label):
     molecules = [molecule_from_smiles(smiles)]

         tf.ragged.constant(pair_indices_list, dtype=tf.int64),
     )
+model = from_pretrained_keras("keras-io/MPNN-for-molecular-property-prediction")
 def predict(smiles, label):
     molecules = [molecule_from_smiles(smiles)]