bho

app.py

@@ -80,20 +80,19 @@ 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")
+    def set_image(image):
+            return image
+
+    # Mostriamo 4 righe da 5 immagini
     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):
-                    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)
-
+            for img, _ in image_list[i:i+5]:
+                img_comp = gr.Image(value=img, interactive=True, show_label=False, show_download_button=False, height=140)
+                img_comp.select(fn=set_image, inputs=img_comp, outputs=image_input)
+    
     submit_btn.click(
         fn=run_prediction,
         inputs=[image_input, model_selector],