update UI

app.py

@@ -693,26 +693,26 @@ def ncut_run(
 
 def _ncut_run(*args, **kwargs):
     n_ret = kwargs.pop("n_ret", 1)
-    # try:
-    #     if torch.cuda.is_available():
-    #         torch.cuda.empty_cache()
+    try:
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
             
-    #     ret = ncut_run(*args, **kwargs)
+        ret = ncut_run(*args, **kwargs)
         
-    #     if torch.cuda.is_available():
-    #         torch.cuda.empty_cache()
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
         
-    #     ret = list(ret)[:n_ret] + [ret[-1]]
-    #     return ret
-    # except Exception as e:
-    #     gr.Error(str(e))
-    #     if torch.cuda.is_available():
-    #         torch.cuda.empty_cache()
-    #     return *(None for _ in range(n_ret)), "Error: " + str(e)
-
-    ret = ncut_run(*args, **kwargs)
-    ret = list(ret)[:n_ret] + [ret[-1]]
-    return ret
+        ret = list(ret)[:n_ret] + [ret[-1]]
+        return ret
+    except Exception as e:
+        gr.Error(str(e))
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+        return *(None for _ in range(n_ret)), "Error: " + str(e)
+
+    # ret = ncut_run(*args, **kwargs)
+    # ret = list(ret)[:n_ret] + [ret[-1]]
+    # return ret
 
 if USE_HUGGINGFACE_ZEROGPU:
     @spaces.GPU(duration=30)
@@ -1105,9 +1105,9 @@ def run_fn(
 
 
 
-def make_input_images_section():
+def make_input_images_section(rows=1, height="auto"):
     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)
+    input_gallery = gr.Gallery(value=None, label="Select images", show_label=True, elem_id="images", columns=[3], rows=[rows], object_fit="contain", height=height, type="pil", show_share_button=False)
     submit_button = gr.Button("🔴 RUN", elem_id="submit_button", variant='primary')
     clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button', variant='stop')
     return input_gallery, submit_button, clear_images_button
@@ -1657,11 +1657,10 @@ with demo:
                 l3_cluster_gallery = gr.Gallery(value=[], label="Recursion #3 Clusters", show_label=True, elem_id="clusters", columns=[2], rows=[4], object_fit="contain", height=600, show_share_button=True, preview=True, interactive=False)
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button = make_input_images_section()
+                input_gallery, submit_button, clear_images_button = make_input_images_section(rows=5, height=1000)
                 dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
                 num_images_slider.value = 100
                 clear_images_button.visible = False
-                logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
                 
             with gr.Column(scale=5, min_width=200):
                 with gr.Accordion("➡️ Recursion config", open=True):
@@ -1681,6 +1680,7 @@ with demo:
                 ] = make_parameters_section()
                 num_eig_slider.visible = False
                 affinity_focal_gamma_slider.visible = False
+                logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
         true_placeholder = gr.Checkbox(label="True placeholder", value=True, elem_id="true_placeholder")
         true_placeholder.visible = False
         false_placeholder = gr.Checkbox(label="False placeholder", value=False, elem_id="false_placeholder")