Update app.py

app.py

@@ -108,13 +108,13 @@ 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=512):
             text_input = gr.Textbox(label="Input Prompt")
             
             orig_img = gr.Image(label="Image Generated by Input Prompt",width=512,show_share_button=False,show_download_button=False)
         with gr.Column():
             start_button = gr.Button("UnlearnDiffAtk!",size='lg')
-        with gr.Column(min_width=260):
+        with gr.Column(min_width=512):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             result_img = gr.Image(label="Image Gnerated by Prompt of UnlearnDiffAtk",width=512,show_share_button=False,show_download_button=False)