added buttons

app.py

@@ -83,16 +83,19 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
     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")
-    with gr.Row():
-        def create_image_selector(image):
-            def select_image():
-                image_input.value = image
-                return gr.update(value=image, visible=True)
-            return select_image
-        
-        for img, path in image_list:
-            img_component = gr.Image(value=img, interactive=True, show_label=False)
-            img_component.select(fn=create_image_selector, inputs=[], outputs=image_input)
+    for i in range(0, len(image_list), 5):
+        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)
 
     submit_btn.click(
         fn=run_prediction,