Update app.py

app.py

@@ -113,10 +113,10 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
     
     for _ in range(repeating_time):
         # try:
-        # model.to(device)
+        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 = 'C'
-        img_list = []
+        # generated_molecule = 'C'
+        # img_list = []
         # Create GIF if img_list is available
         gif_path = None
         if img_list and len(img_list) > 0: