update fpn dino

app.py

@@ -4395,11 +4395,13 @@ with demo:
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
                 input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
+                add_download_button(input_gallery)
                 num_images_slider.value = 30
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information", autofocus=False, autoscroll=False)
                 
             with gr.Column(scale=5, min_width=200):
                 output_gallery = make_output_images_section()
+                add_download_button(output_gallery)
                 # cluster_gallery = gr.Gallery(value=[], label="Clusters", show_label=True, elem_id="clusters", columns=[2], rows=[2], object_fit="contain", height="auto", show_share_button=True, preview=False, interactive=False)
                 [
                     model_dropdown, layer_slider, node_type_dropdown, num_eig_slider, 
@@ -4428,7 +4430,7 @@ with demo:
             outputs=[output_gallery, logging_text],
         )
                         
-    with gr.Tab('Sub-cluster (dev)', visible=False) as sub_cluster_tab:
+    with gr.Tab('Re-color', visible=True) as sub_cluster_tab:
         with gr.Row():
             image_cluster_plot = gr.Image(value=None, label="Image-level clustering", elem_id="image_cluster_plot", interactive=False)
         with gr.Row():
@@ -4970,7 +4972,7 @@ with demo:
         
         hidden_button.change(update_smile, [n_smiles], [n_smiles, hidden_button])
         hidden_tabs = [tab_alignedcut_advanced, tab_model_aligned_advanced, tab_recursivecut_advanced, 
-                        tab_compare_models_advanced, tab_directed_ncut, tab_aligned, tab_lisa, sub_cluster_tab]
+                        tab_compare_models_advanced, tab_directed_ncut, tab_aligned, tab_lisa]
         hidden_button.change(partial(unlock_tabs, n_tab=len(hidden_tabs)), [n_smiles], hidden_tabs)
                 
     with gr.Row():

requirements.txt

@@ -22,4 +22,4 @@ open-clip-torch==2.20.0
 pytorch_lightning==1.9.4
 gradio_image_prompter==0.1.0
 scikit-image==0.24.0
-ncut-pytorch>=1.8.4
+ncut-pytorch>=1.8.8