Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import spaces
+
 from rdkit import Chem
 from rdkit.Chem import Draw
 import numpy as np
@@ -118,7 +120,7 @@ def save_interesting_log(smiles, properties, suggested_properties):
         }
         writer.writerow(log_data)
 
-# @spaces.GPU(duration=60)
+@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