bho

app.py

@@ -80,6 +80,9 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             result_display = gr.Image(label="Model prediction", visible=True)
             error_text = gr.Markdown("", visible=False)
 
+    def updateImage(image: gr.Image):
+        image_input.value = image
+
     with gr.Row():
         gr.Markdown("## Preloaded images to be used for testing the model")
         gr.Markdown("You can use images from the Grand Theft Auto V video game, the Cityscapes dataset or even from Turin")
@@ -87,15 +90,9 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
         with gr.Row():
             for img, path in image_list[i:i+5]:
                 with gr.Column(scale=1):
-                    gr.Image(value=img, show_label=False, show_download_button=False, height=150)
-                    btn = gr.Button(" ", visible=True)
-                    btn.style(height=150, width=150, border_radius="10px", background_color="#f0f0f0", font_size="1px")
-                    def make_on_click(image):
-                        def on_click():
-                            return image
-                        return on_click
-
-                    btn.click(fn=make_on_click(img), outputs=image_input)
+                    btn = gr.Button(value=gr.Image(value=img, show_label=False, show_download_button=False, height=150), 
+                                    visible=True, interactive=True, min_width=150)
+                    btn.click(fn=updateImage, inputs=btn.value, outputs=image_input)
 
     submit_btn.click(
         fn=run_prediction,