upload

app.py

@@ -555,15 +555,16 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                             )
                             seg_image_reset_btn = gr.Button(value="Reset")
                     with gr.Column():
-                        seg_image_output_slider = ImageSlider(
-                            label="Predicted segmentation results",
-                            type="filepath",
-                            show_download_button=True,
-                            show_share_button=True,
-                            interactive=False,
-                            elem_classes="slider",
-                            position=0.25,
-                        )
+                        # seg_image_output_slider = ImageSlider(
+                        #     label="Predicted segmentation results",
+                        #     type="filepath",
+                        #     show_download_button=True,
+                        #     show_share_button=True,
+                        #     interactive=False,
+                        #     elem_classes="slider",
+                        #     position=0.25,
+                        # )
+                        seg_image_output = gr.Gallery(label='Output', show_label=False, elem_id="gallery", columns=[1], height='auto')
                         seg_image_output_files = gr.Files(
                             label="Seg outputs",
                             elem_id="download",
@@ -581,7 +582,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting, pipe_seg, p
                         for name in filenames
                     ],
                     inputs=[seg_image_input],
-                    outputs=[seg_image_output_slider, seg_image_output_files],
+                    outputs=[seg_image_output, seg_image_output_files],
                     cache_examples=False,
                     # directory_name="examples_depth",
                     # cache_examples=False,