Update app.py

app.py

@@ -291,42 +291,42 @@ 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,
-            #             )
+            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)])
@@ -428,38 +428,38 @@ 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_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_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,