Update app.py

app.py

@@ -121,12 +121,10 @@ def process_normal(
 
     np.save(path_out_fp32, depth_pred)
     depth_colored.save(path_out_vis)
-    
-    path_out_16bit = None
 
     return (
-        [path_out_16bit, path_out_vis],
-        [path_out_16bit, path_out_fp32, path_out_vis],
+        [path_out_vis],
+        [path_out_fp32, path_out_vis],
     )
 
 def process_dis(
@@ -156,12 +154,10 @@ def process_dis(
 
     np.save(path_out_fp32, depth_pred)
     depth_colored.save(path_out_vis)
-    
-    path_out_16bit = None
 
     return (
-        [path_out_16bit, path_out_vis],
-        [path_out_16bit, path_out_fp32, path_out_vis],
+        path_out_vis],
+        [path_out_fp32, path_out_vis],
     )
 
 def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
@@ -277,9 +273,9 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                         )
 
                 filenames = []
-                filenames.extend(["anime_%d.jpg" %(i+1) for i in range(7)])
-                filenames.extend(["line_%d.jpg" %(i+1) for i in range(6)])
-                filenames.extend(["real_%d.jpg" %(i+1) for i in range(24)])
+                filenames.extend(["depth_anime_%d.jpg" %(i+1) for i in range(7)])
+                filenames.extend(["depth_line_%d.jpg" %(i+1) for i in range(6)])
+                filenames.extend(["depth_real_%d.jpg" %(i+1) for i in range(24)])
 
                 example_folder = os.path.join(os.path.dirname(__file__), "./images")
                 Examples(
@@ -317,15 +313,16 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                                 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_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 = gr.Gallery(label='Output', show_label=False, elem_id="gallery", columns=[1], height='auto')
                         normal_image_output_files = gr.Files(
                             label="Normal outputs",
                             elem_id="download",
@@ -342,7 +339,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                         for name in filenames
                     ],
                     inputs=[normal_image_input],
-                    outputs=[normal_image_output_slider, normal_image_output_files],
+                    outputs=[normal_image_output, normal_image_output_files],
                     # cache_examples=True,
                     # directory_name="examples_normal",
                     cache_examples=False,
@@ -370,15 +367,16 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                                 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_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 = gr.Gallery(label='Output', show_label=False, elem_id="gallery", columns=[1], height='auto')
                         dis_image_output_files = gr.Files(
                             label="DIS outputs",
                             elem_id="download",
@@ -395,7 +393,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                         for name in filenames
                     ],
                     inputs=[dis_image_input],
-                    outputs=[dis_image_output_slider, dis_image_output_files],
+                    outputs=[dis_image_output, dis_image_output_files],
                     # cache_examples=True,
                     # directory_name="examples_dis",
                     cache_examples=False,
@@ -448,7 +446,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                 normal_image_input,
                 image_processing_res,
             ],
-            outputs=[normal_image_output_slider, normal_image_output_files],
+            outputs=[normal_image_output, normal_image_output_files],
             concurrency_limit=1,
         )
 
@@ -462,7 +460,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
             inputs=[],
             outputs=[
                 normal_image_input,
-                normal_image_output_slider,
+                normal_image_output,
                 normal_image_output_files,
                 image_processing_res,
             ],
@@ -481,7 +479,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
                 dis_image_input,
                 image_processing_res,
             ],
-            outputs=[dis_image_output_slider, dis_image_output_files],
+            outputs=[dis_image_output, dis_image_output_files],
             concurrency_limit=1,
         )
 
@@ -495,7 +493,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis):
             inputs=[],
             outputs=[
                 dis_image_input,
-                dis_image_output_slider,
+                dis_image_output,
                 dis_image_output_files,
                 image_processing_res,
             ],