Update app.py

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.Row():
+        with gr.Column(scale=4, min_width=320):
             text_input = gr.Textbox(label="Input Prompt")
             img1 = gr.Image("images/cheetah.jpg",label="Image")
-        with gr.Column():
+        with gr.Column(scale=1):
             start_button = gr.Button("UnlearnDiffAtk!",size='lg',min_width=10)
-        with gr.Column(min_width=320):
+        with gr.Column(scale=4,min_width=320):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")