vumichien commited on
Commit
32cbfa2
1 Parent(s): 2dbcca5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -95,9 +95,13 @@ def predict(num_mol):
95
  return 'img.png'
96
 
97
  gr.Interface(
98
- predict,
 
 
99
  inputs=[
100
  gr.inputs.Slider(5, 50, label='Number of Molecular Graphs', step=5, default=10),
101
  ],
102
  outputs="image",
103
- ).launch(debug=True)
 
 
 
95
  return 'img.png'
96
 
97
  gr.Interface(
98
+ fn=predict,
99
+ title="Generating molecular graphs by WGAN-GP",
100
+ description = "WGAN-GP with R-GCN for the generation of small molecular graphs 🔬",
101
  inputs=[
102
  gr.inputs.Slider(5, 50, label='Number of Molecular Graphs', step=5, default=10),
103
  ],
104
  outputs="image",
105
+ examples=examples,
106
+ article = "Author: <a href=\"https://huggingface.co/vumichien\">Vu Minh Chien</a>. Based on the keras example from <a href=\"https://keras.io/examples/generative/wgan-graphs/\">Alexander Kensert</a>",
107
+ ).launch(enable_queue=True)