Update app.py

app.py

@@ -80,6 +80,7 @@ with gr.Blocks(css=css) as app:
             with gr.Row():
                 with gr.Column():
                     img_input = gr.Image(type="pil")
+                    img_input_mask = gr.Image(type="pil", visible=False)
                 with gr.Column():
                     options = gr.Dropdown(
                         value=MODELS_NAMES[0],
@@ -99,9 +100,9 @@ with gr.Blocks(css=css) as app:
                     detect_button = gr.Button("Detect leukocytes")
             with gr.Row():
                 example_images = gr.Dataset(
-                    components=[img_input],
+                    components=[img_input, img_input_mask],
                     samples=[
-                        [path.as_posix()]
+                        [path.as_posix(), path.as_posix().replace("_HE", "_mask")]
                         for path in sorted(
                             pathlib.Path("cd45rb_test_imgs").rglob("*_HE.png")
                         )
@@ -143,3 +144,4 @@ with gr.Blocks(css=css) as app:
         fn=set_example_image, inputs=[example_images], outputs=[img_input]
     )
 
+    app.launch(enable_queue=True)