[Update]size to 360

app.py

@@ -80,12 +80,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=300):
+        with gr.Column(min_width=360):
             text_input = gr.Textbox(label="Input Prompt")
             img1 = gr.Image("images/cheetah.jpg",label="Image")
         with gr.Column():
-            start_button = gr.Button("UnlearnDiffAtk!",size='sm')
-        with gr.Column(min_width=300):
+            start_button = gr.Button("UnlearnDiffAtk!",size='lg')
+        with gr.Column(min_width=360):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")