Upload app.py with huggingface_hub

app.py

@@ -54,7 +54,7 @@ def random_properties():
 
 def load_model(model_choice):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    model = load_graph_decoder(device, path=model_choice)
+    model = load_graph_decoder(path=model_choice)
     return (model, device)
 
 # Create a flagged folder if it doesn't exist
@@ -106,6 +106,10 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
                 # print('Before generation, move model to', device)
                 # return generated_molecule, img_list
             # generated_molecule, img_list = generate_func()
+
+            device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+            model.to(device)
+            print('Before generation, move 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