Update app.py

app.py

@@ -101,7 +101,7 @@ def process_image(
 def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
     process_pipe_depth = spaces.GPU(functools.partial(process_image, pipe_depth, mode='depth'))
     process_pipe_normal = spaces.GPU(functools.partial(process_image, pipe_normal, mode='normal'))
-    process_pipe_dis = spaces.GPU(functools.partial(process_image, pipe_dis, mode='dis'))
+    process_pipe_dis = spaces.GPU(functools.partial(process_image, pipe_dis, mode='seg'))
     gradio_theme = gr.themes.Default()
 
     with gr.Blocks(
@@ -180,7 +180,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                             type="filepath",
                         )
                         with gr.Row():
-                            image_submit_btn = gr.Button(
+                            depth_image_submit_btn = gr.Button(
                                 value="Estimate Depth", variant="primary"
                             )
                             image_reset_btn = gr.Button(value="Reset")
@@ -285,7 +285,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                         )
                         with gr.Row():
                             image_submit_btn = gr.Button(
-                                value="Estimate Segmentation.", variant="primary"
+                                value="Estimate Dichotomous Segmentation.", variant="primary"
                             )
                             image_reset_btn = gr.Button(value="Reset")
                         with gr.Accordion("Advanced options", open=False):
@@ -329,7 +329,40 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
 
 
         ### Image tab
-        image_submit_btn.click(
+        depth_image_submit_btn.click(
+            fn=process_image_check,
+            inputs=image_input,
+            outputs=None,
+            preprocess=False,
+            queue=False,
+        ).success(
+            fn=process_pipe_depth,
+            inputs=[
+                image_input,
+                image_processing_res,
+            ],
+            outputs=[image_output_slider, image_output_files],
+            concurrency_limit=1,
+        )
+
+        depth_image_reset_btn.click(
+            fn=lambda: (
+                None,
+                None,
+                None,
+                default_image_processing_res,
+            ),
+            inputs=[],
+            outputs=[
+                image_input,
+                image_output_slider,
+                image_output_files,
+                image_processing_res,
+            ],
+            queue=False,
+        )
+        
+        normal_image_submit_btn.click(
             fn=process_image_check,
             inputs=image_input,
             outputs=None,
@@ -345,7 +378,40 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
             concurrency_limit=1,
         )
 
-        image_reset_btn.click(
+        normal_image_reset_btn.click(
+            fn=lambda: (
+                None,
+                None,
+                None,
+                default_image_processing_res,
+            ),
+            inputs=[],
+            outputs=[
+                image_input,
+                image_output_slider,
+                image_output_files,
+                image_processing_res,
+            ],
+            queue=False,
+        )
+        
+        dis_image_submit_btn.click(
+            fn=process_image_check,
+            inputs=image_input,
+            outputs=None,
+            preprocess=False,
+            queue=False,
+        ).success(
+            fn=process_pipe_dis,
+            inputs=[
+                image_input,
+                image_processing_res,
+            ],
+            outputs=[image_output_slider, image_output_files],
+            concurrency_limit=1,
+        )
+
+        dis_image_reset_btn.click(
             fn=lambda: (
                 None,
                 None,