Update app.py

app.py

@@ -114,9 +114,9 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
     for _ in range(repeating_time):
         # try:
         model.to(device)
-        generated_molecule, img_list = model.generate(properties, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
-        # generated_molecule = 'CC'
-        # img_list = []
+        # generated_molecule, img_list = model.generate(properties, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
+        generated_molecule = 'CC'
+        img_list = []
         # Create GIF if img_list is available
         gif_path = None
         if img_list and len(img_list) > 0: