Update app.py

app.py

@@ -64,7 +64,6 @@ def load_model(model_choice):
     model = GraphDiT(
         model_config_path=model_config_path,
         data_info_path=data_info_path,
-        # model_dtype=torch.float16,
         model_dtype=torch.float32,
     )
     ### test
@@ -113,7 +112,7 @@ 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 = 'CC'
         img_list = []