[Update]Add function of start button

app.py

@@ -27,10 +27,10 @@ from about import (
 )
 
 
-def excute_udiff(diffusion_model_id, concept, attacker):
+def excute_udiff(diffusion_model_id, concept, step):
     print(f"my IP is {myip}, my port is {myport}")
     print(f"my input is diffusion_model_id: {diffusion_model_id}, concept: {concept}, attacker: {attacker}")
-    result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "attacker": attacker})
+    result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "step": step})
     result = result.text[1:-1]
     
     return result
@@ -93,7 +93,7 @@ with gr.Blocks(css=custom_css) as demo:
     #         ["CompVis/stable-diffusion-v1-4", "nudity", "text_grad"]
     #     ], inputs=[diffusion_model_id, concept, attacker])
 
-    # start_button.click(fn=excute_udiff, inputs=[diffusion_model_id, concept, attacker], outputs=result, api_name="udiff")
+    start_button.click(fn=excute_udiff, inputs=[drop_model, drop, shown_columns_step], outputs=result, api_name="udiff")
 
 
 # demo.queue(default_enabled=False, api_open=False, max_size=5).launch(debug=True, show_api=False)