Update app.py

app.py

@@ -83,8 +83,8 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Column(min_width=300):
             text_input = gr.Textbox(label="Input Prompt")
             img1 = gr.Image("images/cheetah.jpg",label="Image")
-        with gr.Column(scale=4, min_width=40):
-            start_button = gr.Button("UnlearnDiffAtk!")
+        with gr.Column():
+            start_button = gr.Button("UnlearnDiffAtk!",size='lg')
         with gr.Column(min_width=300):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
             img2 = gr.Image("images/cheetah.jpg",label="After Defense")