[Update]change width size

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=260):
+        with gr.Column(min_width=300):
             text_input = gr.Textbox(label="Input Prompt")
             img1 = gr.Image("images/cheetah.jpg",label="Image")
-        with gr.Column(min_width=60):
+        with gr.Column(scale=4, min_width=40):
             start_button = gr.Button("UnlearnDiffAtk!")
-        with gr.Column(min_width=260):
+        with gr.Column(min_width=300):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")