Update app.py

app.py

@@ -100,6 +100,7 @@ def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_ti
     num_nodes = None if num_nodes == 0 else num_nodes
     
     for _ in range(repeating_time):
+        print('repeat times:', _)
         try:
             model.to(device)
             generated_molecule, img_list = model.generate(properties, guide_scale=guidance_scale, num_nodes=num_nodes, number_chain_steps=num_chain_steps)
@@ -275,7 +276,7 @@ with gr.Blocks(title="Polymer Design with GraphDiT") as iface:
     with gr.Row():
         guidance_scale = gr.Slider(1, 3, value=2, label="Guidance Scale from Properties")
         num_nodes = gr.Slider(0, 50, step=1, value=0, label="Number of Nodes (0 for Random, Larger Graphs Take More Time)")
-        repeating_time = gr.Slider(1, 10, step=1, value=3, label="Repetition Until Success")
+        repeating_time = gr.Slider(1, 10, step=1, value=5, label="Repetition Until Success")
         num_chain_steps = gr.Slider(0, 499, step=1, value=50, label="Number of Diffusion Steps to Visualize (Larger Numbers Take More Time)")
         fps = gr.Slider(0.25, 10, step=0.25, value=5, label="Frames Per Second")