Update app.py
Browse files
app.py
CHANGED
@@ -114,7 +114,7 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
|
|
114 |
for _ in range(repeating_time):
|
115 |
# try:
|
116 |
# model.to(device)
|
117 |
-
|
118 |
generated_molecule = 'C'
|
119 |
img_list = []
|
120 |
# Create GIF if img_list is available
|
|
|
114 |
for _ in range(repeating_time):
|
115 |
# try:
|
116 |
# model.to(device)
|
117 |
+
generated_molecule, img_list = model.generate(properties, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
|
118 |
generated_molecule = 'C'
|
119 |
img_list = []
|
120 |
# Create GIF if img_list is available
|