Update app.py

app.py

@@ -481,9 +481,7 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting):
                         )
 
                 filenames = []
-                filenames.extend(["matting_%d.jpg" %(i+1) for i in range(10)])
-                # example_folder = "images"
-                # print('line 396', __file__)
+                filenames.extend(["matting_%d.jpg" %(i+1) for i in range(4)])
                 example_folder = os.path.join(os.path.dirname(__file__), "matting_images")
                 # print(example_folder)
                 Examples(
@@ -492,8 +490,8 @@ def run_demo_server(pipe_depth, pipe_normal, pipe_dis, pipe_matting):
                         os.path.join(example_folder, name)
                         for name in filenames
                     ],
-                    inputs=[dis_image_input],
-                    outputs=[dis_image_output, dis_image_output_files],
+                    inputs=[matting_image_input],
+                    outputs=[matting_image_output, matting_image_output_files],
                     # cache_examples=True,
                     directory_name="images_cache",
                     cache_examples=False,