[Update]image position

app.py

@@ -40,14 +40,12 @@ with gr.Blocks() as demo:
         
     with gr.Row() as attack:
         text_input = gr.Textbox(label="Prompt")
-        with gr.Row():
-            with gr.Column(min_width=260):
-                
-                img1 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
-            with gr.Column():
-                start_button = gr.Button("AdvUnlearn",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 Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
+    with gr.Row():
+        with gr.Column(min_width=260):
+            img1 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
+        with gr.Column():
+            start_button = gr.Button("AdvUnlearn",size='lg')
+        with gr.Column(min_width=260):
+            img2 = gr.Image("images/cheetah.jpg",label="Image Generated without AdvUnlearn",width=260,show_share_button=False,show_download_button=False)
             
 demo.launch()