Update app.py

app.py

@@ -82,6 +82,18 @@ with gr.Blocks(css=css) as app:
                 with gr.Column():
                     img_input = gr.Image(type="pil")
                     img_input_mask = gr.Image(type="pil", visible=False)
+                with gr.Column():
+                    example_images = gr.Dataset(
+                        components=[img_input, img_input_mask],
+                        samples=[
+                            [path.as_posix(), path.as_posix().replace("_HE", "_mask")]
+                            for path in sorted(
+                                pathlib.Path("cd45rb_test_imgs").rglob("*_HE.png")
+                            )
+                        ],
+                        samples_per_page=2,
+                    )
+            with gr.Row():
                 with gr.Column():
                     options = gr.Dropdown(
                         value=MODELS_NAMES[0],
@@ -89,27 +101,15 @@ with gr.Blocks(css=css) as app:
                         label="Select an object detection model",
                         show_label=True,
                     )
-                    with gr.Row():
-                        with gr.Column():
-                            slider_input = gr.Slider(
-                                minimum=0.2, maximum=1, value=0.7, label="Prediction threshold"
-                            )
-                        with gr.Column():
-                            display_mask = gr.Checkbox(
-                                label="Display masks", default=False
-                            )
-                    detect_button = gr.Button("Detect leukocytes")
-            with gr.Row():
-                example_images = gr.Dataset(
-                    components=[img_input, img_input_mask],
-                    samples=[
-                        [path.as_posix(), path.as_posix().replace("_HE", "_mask")]
-                        for path in sorted(
-                            pathlib.Path("cd45rb_test_imgs").rglob("*_HE.png")
-                        )
-                    ],
-                    samples_per_page=2,
-                )
+                with gr.Column():
+                    slider_input = gr.Slider(
+                        minimum=0.2, maximum=1, value=0.7, label="Prediction threshold"
+                    )
+                with gr.Column():
+                    display_mask = gr.Checkbox(
+                        label="Display masks", default=False
+                    )
+                detect_button = gr.Button("Detect leukocytes")
             with gr.Row():
                 with gr.Column():
                     gr.Markdown(