liuganghuggingface commited on
Commit
fd8ad98
·
verified ·
1 Parent(s): d7a9f21

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, device=device, 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
 
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