Update app.py

app.py

@@ -37,10 +37,10 @@ def process_image_from_binary(img_stream):
     
     return img
 
-def excute_udiff(diffusion_model_id, concept, steps):
+def excute_udiff(diffusion_model_id, concept, steps, attack_id):
     print(f"my IP is {myip}, my port is {myport}")
     print(f"my input is diffusion_model_id: {diffusion_model_id}, concept: {concept}, steps: {steps}")
-    response = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "steps": steps})
+    response = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "steps": steps, "attack_id": attack_id})
     print(f"result: {response}")
     # result = result.text[1:-1]
     prompt1 = ""
@@ -116,7 +116,7 @@ with gr.Blocks(css=custom_css) as demo:
             result_img = gr.Image(label="Image Gnerated by Prompt of UnlearnDiffAtk",width=260,show_share_button=False,show_download_button=False)
             
 
-    start_button.click(fn=excute_udiff, inputs=[drop_model, drop, shown_columns_step, int(atk_idx)], outputs=[text_input, text_ouput, orig_img, result_img], api_name="udiff")
+    start_button.click(fn=excute_udiff, inputs=[drop_model, drop, shown_columns_step, atk_idx], outputs=[text_input, text_ouput, orig_img, result_img], api_name="udiff")
 
 
 demo.queue().launch(server_name='0.0.0.0')