Upload app.py with huggingface_hub

app.py

@@ -80,6 +80,7 @@ def save_interesting_log(smiles, properties, suggested_properties):
         }
         writer.writerow(log_data)
 
+@spaces.GPU(duration=60)
 def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):
     model, device = model_state
     
@@ -98,15 +99,13 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
     
     for _ in range(repeating_time):
         try:
-            @spaces.GPU(duration=60)
-            def 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)
-                return generated_molecule, img_list
-
-            generated_molecule, img_list = generate_func()
+            # 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()
+            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
             gif_path = None
@@ -347,4 +346,4 @@ with gr.Blocks(title="Polymer Design with GraphDiT") as iface:
 # Launch the interface
 if __name__ == "__main__":
     # iface.launch(share=True)
-    iface.launch()
+    iface.launch(share=False)