improve UI

app.py

@@ -455,7 +455,7 @@ def run_fn(
 def make_input_images_section():
     gr.Markdown('### Input Images')
     input_gallery = gr.Gallery(value=None, 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("🔴RUN", elem_id="submit_button")
+    submit_button = gr.Button("🔴 RUN", elem_id="submit_button")
     clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button')
     return input_gallery, submit_button, clear_images_button
 
@@ -465,7 +465,7 @@ def make_input_video_section():
     gr.Markdown('_image backbone model is used to extract features from each frame, NCUT is computed on all frames_')
     # max_frames_number = gr.Number(100, label="Max frames", elem_id="max_frames")
     max_frames_number = gr.Slider(1, 200, step=1, label="Max frames", value=100, elem_id="max_frames")
-    submit_button = gr.Button("🔴RUN", elem_id="submit_button")
+    submit_button = gr.Button("🔴 RUN", elem_id="submit_button")
     clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button')
     return input_gallery, submit_button, clear_images_button, max_frames_number
 
@@ -478,6 +478,10 @@ def make_example_images_section():
         fn=lambda: gr.update(visible=False),
         outputs=example_gallery
     )
+    hide_button.click(
+        fn=lambda: gr.update(visible=False),
+        outputs=hide_button
+    )
     return load_images_button, example_gallery, hide_button
 
 def make_example_video_section():
@@ -615,8 +619,12 @@ def make_parameters_section():
             perplexity_slider, n_neighbors_slider, min_dist_slider, 
             sampling_method_dropdown]
 
-with gr.Blocks() as demo:
-
+demo = gr.Blocks(
+    theme=gr.themes.Base(spacing_size='md', text_size='lg', primary_hue='blue', neutral_hue='slate', secondary_hue='pink'),
+    fill_width=False,
+    title="ncut-pytorch",
+)
+with demo:
     with gr.Tab('AlignedCut'):
 
         with gr.Row():
@@ -829,7 +837,7 @@ with gr.Blocks() as demo:
             with gr.Column(scale=5, min_width=200) as col:
                 gr.Markdown(f'### Output Images')
                 output_gallery = gr.Gallery(value=[], label="NCUT Embedding", show_label=False, elem_id=f"ncut{i_model}", columns=[3], rows=[1], object_fit="contain", height="auto")
-                submit_button = gr.Button("🔴RUN", elem_id=f"submit_button{i_model}")
+                submit_button = gr.Button("🔴 RUN", elem_id=f"submit_button{i_model}")
                 [
                     model_dropdown, layer_slider, node_type_dropdown, num_eig_slider, 
                     affinity_focal_gamma_slider, num_sample_ncut_slider, knn_ncut_slider,