update UI

app.py

@@ -136,12 +136,10 @@ def get_random_path(length=10):
     return path
 
 default_images = ['./images/image_0.jpg', './images/image_1.jpg', './images/image_2.jpg', './images/image_3.jpg', './images/image_5.jpg']
-# default_outputs = ['./images/ncut_0.jpg', './images/ncut_1.jpg', './images/ncut_2.jpg', './images/ncut_3.jpg', './images/ncut_5.jpg']
 default_outputs = ['./images/image-1.webp', './images/image-2.webp', './images/image-3.webp', './images/image-4.webp', './images/image-5.webp']
-default_outputs_independent = ['./images/ncut_0_independent.jpg', './images/ncut_1_independent.jpg', './images/ncut_2_independent.jpg', './images/ncut_3_independent.jpg', './images/ncut_5_independent.jpg']
+default_outputs_independent = ['./images/image-6.webp', './images/image-7.webp', './images/image-8.webp', './images/image-9.webp', './images/image-10.webp']
 
 downscaled_images = ['./images/image_0_small.jpg', './images/image_1_small.jpg', './images/image_2_small.jpg', './images/image_3_small.jpg', './images/image_5_small.jpg']
-# downscaled_outputs = ['./images/ncut_0_small.jpg', './images/ncut_1_small.jpg', './images/ncut_2_small.jpg', './images/ncut_3_small.jpg', './images/ncut_5_small.jpg']
 downscaled_outputs = default_outputs
 
 example_items = downscaled_images[:3] + downscaled_outputs[:3]
@@ -433,8 +431,8 @@ def make_example_video_section():
     load_video_button = gr.Button("Load Example", elem_id="load-video-button")
     return load_video_button
 
-def make_dataset_images_section():
-    with gr.Accordion("➡️ Click to expand: Load from dataset", open=False):
+def make_dataset_images_section(open=False):
+    with gr.Accordion("➡️ Click to expand: Load from dataset", open=open):
         dataset_names = [
             'UCSC-VLAA/Recap-COCO-30K',
             'nateraw/pascal-voc-2012',
@@ -469,7 +467,7 @@ def make_dataset_images_section():
     
 def make_output_images_section():
     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")
+    output_gallery = gr.Gallery(value=[], label="NCUT Embedding", show_label=False, elem_id="ncut", columns=[3], rows=[1], object_fit="contain", height="auto", show_download_button=True
     return output_gallery
 
 def make_parameters_section():
@@ -735,7 +733,7 @@ with gr.Blocks() as demo:
                 load_images_button, example_gallery, hide_button = make_example_images_section()
                 example_gallery.visible = False
                 hide_button.visible = False
-                dataset_dropdown, num_images_slider, random_seed_slider, load_dataset_button = make_dataset_images_section()
+                dataset_dropdown, num_images_slider, random_seed_slider, load_dataset_button = make_dataset_images_section(open=True)
                 load_images_button.click(lambda x: default_images, outputs=input_gallery)
                 
             with gr.Column(scale=5, min_width=200):