Upload app.py with huggingface_hub

app.py

@@ -58,6 +58,8 @@ def random_properties():
 model_all = load_graph_decoder(path='model_all')
 model_labeled = load_graph_decoder(path='model_labeled')
     # return (model, device)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print('device')
 
 # Create a flagged folder if it doesn't exist
 flagged_folder = "flagged"
@@ -110,7 +112,7 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
                 # return generated_molecule, img_list
             # generated_molecule, img_list = generate_func()
             # print('Before generation, move model to', device)
-            model.to(device)
+            # model.to(device)
             generated_molecule, img_list = model.generate(properties, device=device, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
 
             # Create GIF if img_list is available