update ui

app.py

@@ -137,6 +137,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column(scale=5, min_width=200):
+            gr.Markdown('### Input Images')
             input_gallery = gr.Gallery(value=[], label="Select images", show_label=False, elem_id="images", columns=[3], rows=[1], object_fit="contain", height="auto", type="pil", show_share_button=False)
             submit_button = gr.Button("🔴Submit", elem_id="submit_button")
             clear_images_button = gr.Button("🗑️Clear Images")
@@ -146,6 +147,7 @@ with gr.Blocks() as demo:
             gr.Gallery(value=downscaled_images[:3] + downscaled_outputs[:3], label="Example Set A", show_label=False, columns=[3], rows=[2], object_fit="scale-down", height="200px", show_share_button=False)
         
         with gr.Column(scale=5, min_width=200):
+            gr.Markdown('### Output Images')
             output_gallery = gr.Gallery(value=[], label="NCUT Embedding", show_label=False, elem_id="ncut", columns=[3], rows=[1], object_fit="contain", height="auto")
             model_dropdown = gr.Dropdown(["SAM(sam_vit_b)", "MobileSAM", "DiNO(dinov2_vitb14_reg)", "CLIP(openai/clip-vit-base-patch16)"], label="Model", value="SAM(sam_vit_b)", elem_id="model_name")
             layer_slider = gr.Slider(0, 11, step=1, label="Layer", value=11, elem_id="layer")