[Update]Add image width

app.py

@@ -81,16 +81,12 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row() as attack:
         with gr.Column(min_width=260):
             text_input = gr.Textbox(label="Input Prompt")
-            img1 = gr.Image("images/cheetah.jpg",label="Image Generated by Input Prompt",
-                           show_share_button=False,
-                           show_download_button=False)
+            img1 = gr.Image("images/cheetah.jpg",label="Image Generated by Input Prompt",width=260,show_share_button=False,show_download_button=False)
         with gr.Column():
             start_button = gr.Button("UnlearnDiffAtk!",size='lg')
         with gr.Column(min_width=260):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
-                img2 = gr.Image("images/cheetah.jpg",label="Image Gnerated by Prompt of UnlearnDiffAtk",
-                                show_share_button=False,
-                                   show_download_button=False)
+            img2 = gr.Image("images/cheetah.jpg",label="Image Gnerated by Prompt of UnlearnDiffAtk",width=260,show_share_button=False,show_download_button=False)
             
     # with gr.Column():
     #     gr.Examples(examples=[