TRIED FIXING SELECTION

app.py

@@ -12,8 +12,9 @@ city_image_dir = "./preloadedImages/cityScapes"
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 
-def load_example(example_img):
-    return gr.update(value=example_img)
+def load_example(index:int=0, useGta:bool=True)-> gr.Image:
+    example_img = loadPreloadedImages(gta_image_dir if useGta else city_image_dir)
+    return gr.update(value=example_img[index if 0 <= index < len(example_img) else 0], visible=True)
 
 
 # %% prediction on an image
@@ -75,21 +76,21 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
         gr.Markdown("## Preloaded GTA V images to be used for testing the model")
     with gr.Row():
         gta_gallery = gr.Gallery(value=loadPreloadedImages(gta_image_dir),
-            label="GTA V Examples", show_label=False, columns=5,
-            rows=1, height=200, type="pil", allow_preview=False
+            label="GTA V Examples", show_label=False, columns=5, type="index",
+            rows=1, height=200, allow_preview=False
         )
 
     with gr.Row():
         gr.Markdown("## Preloaded Cityscapes images to be used for testing the model")
     with gr.Row():
         city_gallery = gr.Gallery(value=loadPreloadedImages(city_image_dir),
-            label="Cityscapes Examples", show_label=False, columns=5,
-            rows=1, height=256, type="pil", allow_preview=False
+            label="Cityscapes Examples", show_label=False, columns=5, type="index",
+            rows=1, height=256, allow_preview=False
     )
     
-    gta_gallery.select(fn=load_example, inputs=[gta_gallery], outputs=[image_input])
-    city_gallery.select(fn=load_example, inputs=[city_gallery], outputs=[image_input])
-    
+    gta_gallery.select(fn=load_example, inputs=[gta_gallery, True], outputs=[predict(image_input)])
+    city_gallery.select(fn=load_example, inputs=[city_gallery, False], outputs=[predict(image_input)])
+
     submit_btn.click(
         fn=run_prediction,
         inputs=[image_input, model_selector],