Update app.py

app.py

@@ -225,107 +225,107 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                     directory_name="examples_image",
                 )
             
-            with gr.Tab("Normal"):
-                with gr.Row():
-                    with gr.Column():
-                        normal_image_input = gr.Image(
-                            label="Input Image",
-                            type="filepath",
-                        )
-                        with gr.Row():
-                            normal_image_submit_btn = gr.Button(
-                                value="Estimate Normal", variant="primary"
-                            )
-                            normal_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
-                    with gr.Column():
-                        normal_image_output_slider = ImageSlider(
-                            label="Predicted surface normal",
-                            type="filepath",
-                            show_download_button=True,
-                            show_share_button=True,
-                            interactive=False,
-                            elem_classes="slider",
-                            position=0.25,
-                        )
-                        normal_image_output_files = gr.Files(
-                            label="Normal outputs",
-                            elem_id="download",
-                            interactive=False,
-                        )
-
-                filenames = []
-                filenames.extend(["%d.jpg" %(i+1) for i in range(10)])
-                Examples(
-                    fn=process_pipe_normal,
-                    examples=[
-                        os.path.join("images", "normal", name)
-                        for name in filenames
-                    ],
-                    inputs=[normal_image_input],
-                    outputs=[normal_image_output_slider, normal_image_output_files],
-                    cache_examples=True,
-                    directory_name="examples_image",
-                )
+            # with gr.Tab("Normal"):
+            #     with gr.Row():
+            #         with gr.Column():
+            #             normal_image_input = gr.Image(
+            #                 label="Input Image",
+            #                 type="filepath",
+            #             )
+            #             with gr.Row():
+            #                 normal_image_submit_btn = gr.Button(
+            #                     value="Estimate Normal", variant="primary"
+            #                 )
+            #                 normal_image_reset_btn = gr.Button(value="Reset")
+            #             with gr.Accordion("Advanced options", open=False):
+            #                 image_processing_res = gr.Radio(
+            #                     [
+            #                         ("Native", 0),
+            #                         ("Recommended", 768),
+            #                     ],
+            #                     label="Processing resolution",
+            #                     value=default_image_processing_res,
+            #                 )
+            #         with gr.Column():
+            #             normal_image_output_slider = ImageSlider(
+            #                 label="Predicted surface normal",
+            #                 type="filepath",
+            #                 show_download_button=True,
+            #                 show_share_button=True,
+            #                 interactive=False,
+            #                 elem_classes="slider",
+            #                 position=0.25,
+            #             )
+            #             normal_image_output_files = gr.Files(
+            #                 label="Normal outputs",
+            #                 elem_id="download",
+            #                 interactive=False,
+            #             )
+
+            #     filenames = []
+            #     filenames.extend(["%d.jpg" %(i+1) for i in range(10)])
+            #     Examples(
+            #         fn=process_pipe_normal,
+            #         examples=[
+            #             os.path.join("images", "normal", name)
+            #             for name in filenames
+            #         ],
+            #         inputs=[normal_image_input],
+            #         outputs=[normal_image_output_slider, normal_image_output_files],
+            #         cache_examples=True,
+            #         directory_name="examples_image",
+            #     )
                 
-            with gr.Tab("Dichotomous Segmentation"):
-                with gr.Row():
-                    with gr.Column():
-                        dis_image_input = gr.Image(
-                            label="Input Image",
-                            type="filepath",
-                        )
-                        with gr.Row():
-                            dis_image_submit_btn = gr.Button(
-                                value="Estimate Dichotomous Segmentation.", variant="primary"
-                            )
-                            dis_image_reset_btn = gr.Button(value="Reset")
-                        with gr.Accordion("Advanced options", open=False):
-                            image_processing_res = gr.Radio(
-                                [
-                                    ("Native", 0),
-                                    ("Recommended", 768),
-                                ],
-                                label="Processing resolution",
-                                value=default_image_processing_res,
-                            )
-                    with gr.Column():
-                        dis_image_output_slider = ImageSlider(
-                            label="Predicted dichotomous image segmentation",
-                            type="filepath",
-                            show_download_button=True,
-                            show_share_button=True,
-                            interactive=False,
-                            elem_classes="slider",
-                            position=0.25,
-                        )
-                        dis_image_output_files = gr.Files(
-                            label="DIS outputs",
-                            elem_id="download",
-                            interactive=False,
-                        )
-
-                filenames = []
-                filenames.extend(["%d.jpg" %(i+1) for i in range(10)])
-                Examples(
-                    fn=process_pipe_dis,
-                    examples=[
-                        os.path.join("images", "dis", name)
-                        for name in filenames
-                    ],
-                    inputs=[dis_image_input],
-                    outputs=[dis_image_output_slider, dis_image_output_files],
-                    cache_examples=True,
-                    directory_name="examples_image",
-                )
+            # with gr.Tab("Dichotomous Segmentation"):
+            #     with gr.Row():
+            #         with gr.Column():
+            #             dis_image_input = gr.Image(
+            #                 label="Input Image",
+            #                 type="filepath",
+            #             )
+            #             with gr.Row():
+            #                 dis_image_submit_btn = gr.Button(
+            #                     value="Estimate Dichotomous Segmentation.", variant="primary"
+            #                 )
+            #                 dis_image_reset_btn = gr.Button(value="Reset")
+            #             with gr.Accordion("Advanced options", open=False):
+            #                 image_processing_res = gr.Radio(
+            #                     [
+            #                         ("Native", 0),
+            #                         ("Recommended", 768),
+            #                     ],
+            #                     label="Processing resolution",
+            #                     value=default_image_processing_res,
+            #                 )
+            #         with gr.Column():
+            #             dis_image_output_slider = ImageSlider(
+            #                 label="Predicted dichotomous image segmentation",
+            #                 type="filepath",
+            #                 show_download_button=True,
+            #                 show_share_button=True,
+            #                 interactive=False,
+            #                 elem_classes="slider",
+            #                 position=0.25,
+            #             )
+            #             dis_image_output_files = gr.Files(
+            #                 label="DIS outputs",
+            #                 elem_id="download",
+            #                 interactive=False,
+            #             )
+
+            #     filenames = []
+            #     filenames.extend(["%d.jpg" %(i+1) for i in range(10)])
+            #     Examples(
+            #         fn=process_pipe_dis,
+            #         examples=[
+            #             os.path.join("images", "dis", name)
+            #             for name in filenames
+            #         ],
+            #         inputs=[dis_image_input],
+            #         outputs=[dis_image_output_slider, dis_image_output_files],
+            #         cache_examples=True,
+            #         directory_name="examples_image",
+            #     )
 
 
         ### Image tab
@@ -362,71 +362,71 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
             queue=False,
         )
         
-        normal_image_submit_btn.click(
-            fn=process_image_check,
-            inputs=normal_image_input,
-            outputs=None,
-            preprocess=False,
-            queue=False,
-        ).success(
-            fn=process_pipe_normal,
-            inputs=[
-                normal_image_input,
-                image_processing_res,
-            ],
-            outputs=[normal_image_output_slider, normal_image_output_files],
-            concurrency_limit=1,
-        )
-
-        normal_image_reset_btn.click(
-            fn=lambda: (
-                None,
-                None,
-                None,
-                default_image_processing_res,
-            ),
-            inputs=[],
-            outputs=[
-                normal_image_input,
-                normal_image_output_slider,
-                normal_image_output_files,
-                image_processing_res,
-            ],
-            queue=False,
-        )
+        # normal_image_submit_btn.click(
+        #     fn=process_image_check,
+        #     inputs=normal_image_input,
+        #     outputs=None,
+        #     preprocess=False,
+        #     queue=False,
+        # ).success(
+        #     fn=process_pipe_normal,
+        #     inputs=[
+        #         normal_image_input,
+        #         image_processing_res,
+        #     ],
+        #     outputs=[normal_image_output_slider, normal_image_output_files],
+        #     concurrency_limit=1,
+        # )
+
+        # normal_image_reset_btn.click(
+        #     fn=lambda: (
+        #         None,
+        #         None,
+        #         None,
+        #         default_image_processing_res,
+        #     ),
+        #     inputs=[],
+        #     outputs=[
+        #         normal_image_input,
+        #         normal_image_output_slider,
+        #         normal_image_output_files,
+        #         image_processing_res,
+        #     ],
+        #     queue=False,
+        # )
         
-        dis_image_submit_btn.click(
-            fn=process_image_check,
-            inputs=dis_image_input,
-            outputs=None,
-            preprocess=False,
-            queue=False,
-        ).success(
-            fn=process_pipe_dis,
-            inputs=[
-                dis_image_input,
-                image_processing_res,
-            ],
-            outputs=[dis_image_output_slider, dis_image_output_files],
-            concurrency_limit=1,
-        )
-
-        dis_image_reset_btn.click(
-            fn=lambda: (
-                None,
-                None,
-                None,
-                default_image_processing_res,
-            ),
-            inputs=[],
-            outputs=[
-                dis_image_input,
-                dis_image_output_slider,
-                dis_image_output_files,
-                image_processing_res,
-            ],
-            queue=False,
-        )
+        # dis_image_submit_btn.click(
+        #     fn=process_image_check,
+        #     inputs=dis_image_input,
+        #     outputs=None,
+        #     preprocess=False,
+        #     queue=False,
+        # ).success(
+        #     fn=process_pipe_dis,
+        #     inputs=[
+        #         dis_image_input,
+        #         image_processing_res,
+        #     ],
+        #     outputs=[dis_image_output_slider, dis_image_output_files],
+        #     concurrency_limit=1,
+        # )
+
+        # dis_image_reset_btn.click(
+        #     fn=lambda: (
+        #         None,
+        #         None,
+        #         None,
+        #         default_image_processing_res,
+        #     ),
+        #     inputs=[],
+        #     outputs=[
+        #         dis_image_input,
+        #         dis_image_output_slider,
+        #         dis_image_output_files,
+        #         image_processing_res,
+        #     ],
+        #     queue=False,
+        # )
 
         ### Server launch