[update]Change to 320

app.py

@@ -1,4 +1,3 @@
-import gradio as gr
 
 import gradio as gr
 import os
@@ -80,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=360):
+        with gr.Column(min_width=320):
             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=360):
+            start_button = gr.Button("UnlearnDiffAtk!",size='lg',min_width=10)
+        with gr.Column(min_width=320):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")