Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
@@ -103,7 +103,7 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
|
|
103 |
for _ in range(repeating_time):
|
104 |
# try:
|
105 |
model.to(device)
|
106 |
-
generated_molecule, img_list = model.generate(properties,
|
107 |
|
108 |
# Create GIF if img_list is available
|
109 |
gif_path = None
|
|
|
103 |
for _ in range(repeating_time):
|
104 |
# try:
|
105 |
model.to(device)
|
106 |
+
generated_molecule, img_list = model.generate(properties, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
|
107 |
|
108 |
# Create GIF if img_list is available
|
109 |
gif_path = None
|