Update app.py

app.py

@@ -59,8 +59,8 @@ def load_model(model_choice):
     #### test
     from graph_decoder.diffusion_model import GraphDiT
     
-    model_config_path = f"model_labeled/config.yaml"
-    data_info_path = f"model_labeled/data.meta.json"
+    model_config_path = f"{model_labeled}/config.yaml"
+    data_info_path = f"{model_labeled}/data.meta.json"
     model = GraphDiT(
         model_config_path=model_config_path,
         data_info_path=data_info_path,
@@ -113,10 +113,8 @@ 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 = 'C'
-        img_list = []
+        model.to(device)
+        generated_molecule, img_list = model.generate(properties, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
         # Create GIF if img_list is available
         gif_path = None
         if img_list and len(img_list) > 0: