Update app.py

app.py

@@ -81,9 +81,9 @@ def save_interesting_log(smiles, properties, suggested_properties):
         }
         writer.writerow(log_data)
 
-# @spaces.GPU(duration=60)
-@spaces.GPU
+@spaces.GPU(duration=60)
 def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):
+    print('in generate graph', torch.device("cuda" if torch.cuda.is_available() else "cpu"))
     model, device = model_state
     
     properties = [CH4, CO2, H2, N2, O2]
@@ -101,13 +101,6 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
     
     for _ in range(repeating_time):
         try:
-            # def generate_func():
-                # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-                # model.to(device)
-                # print('Before generation, move model to', device)
-                # return generated_molecule, img_list
-            # generated_molecule, img_list = generate_func()
-
             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)