[Update]caption images

app.py

@@ -81,11 +81,11 @@ 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")
+            img1 = gr.Image("images/cheetah.jpg",label="Image Generated by Input Prompt")
         with gr.Column():
-            start_button = gr.Button("UnlearnDiffAtk!",size='sm')
+            start_button = gr.Button("UnlearnDiffAtk!",size='lg')
         with gr.Column(min_width=260):
-            text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
+            text_ouput = gr.Textbox(label="Prompt Genetated by Prompt of UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")
             
     # with gr.Column():