[Update]Change to 260

app.py

@@ -79,12 +79,12 @@ with gr.Blocks(css=custom_css) as demo:
                             step=1, label="Attack Steps", info="Choose between 0 and 100",
                             interactive=True,)
     with gr.Row() as attack:
-        with gr.Column(min_width=320):
+        with gr.Column(min_width=260):
             text_input = gr.Textbox(label="Input Prompt")
             img1 = gr.Image("images/cheetah.jpg",label="Image")
         with gr.Column():
-            start_button = gr.Button("UnlearnDiffAtk!",size='lg')
-        with gr.Column(min_width=320):
+            start_button = gr.Button("UnlearnDiffAtk!",size='sm')
+        with gr.Column(min_width=260):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")