Spaces:

liuganghuggingface
/

Polymer-Design-With-GraphDiT

Sleeping

liuganghuggingface commited on Oct 7, 2024

Commit

1f5a5fc

verified ·

1 Parent(s): d53fd91

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,7 @@ def test(n):
     # print('1 ',torch.Tensor([1]).cuda())
     return 0
-test(n-1)
 @spaces.GPU
 def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):

     # print('1 ',torch.Tensor([1]).cuda())
     return 0
+test(n)
 @spaces.GPU
 def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):