Update app.py

app.py

@@ -96,6 +96,8 @@ with gr.Blocks(css=custom_css) as demo:
             # attacker = gr.Textbox(label='attacker')
 
             # start_button = gr.Button("Attack!")
+        with gr.Column():
+            atk_idx = gr.Textbox(label="attack index")
 
         with gr.Column():
              shown_columns_step = gr.Slider(
@@ -108,14 +110,13 @@ with gr.Blocks(css=custom_css) as demo:
             
             orig_img = gr.Image(label="Image Generated by Input Prompt",width=260,show_share_button=False,show_download_button=False)
         with gr.Column():
-            atk_idx = gr.Textbox(label="attack index")
             start_button = gr.Button("UnlearnDiffAtk!",size='lg')
         with gr.Column(min_width=260):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             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, 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, int(atk_idx)], outputs=[text_input, text_ouput, orig_img, result_img], api_name="udiff")
 
 
 demo.queue().launch(server_name='0.0.0.0')