chore: better ui

app.py

@@ -142,10 +142,8 @@ TKG-DM🥚🍚: Training-free Chroma Key Content Generation Diffusion Model
             with gr.Column():
                 prompt = gr.Text(
                     label="Prompt",
-                    show_label=False,
-                    max_lines=1,
+                    max_lines=4,
                     placeholder="Enter your prompt",
-                    container=False,
                 )
                 negative_prompt = gr.Textbox(
                     label="Negative prompt",
@@ -207,9 +205,9 @@ TKG-DM🥚🍚: Training-free Chroma Key Content Generation Diffusion Model
                         )
 
             with gr.Column():
-                run_button = gr.Button("Run", scale=0, variant="primary")
-                result_w_tkg = gr.Image(label="Result", show_label=False)
-                result_wo_tkg = gr.Image(label="Result", show_label=False)
+                run_button = gr.Button("Generate", variant="primary")
+                result_w_tkg = gr.Image(label="with TKG")
+                result_wo_tkg = gr.Image(label="without TKG")