TRIED FIXING SELECTION

app.py

@@ -88,8 +88,8 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             rows=1, height=256, allow_preview=False
     )
     
-    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)])
+    gta_gallery.select(fn=load_example, inputs=[gta_gallery, True], outputs=[image_input])
+    city_gallery.select(fn=load_example, inputs=[city_gallery, False], outputs=[image_input])
 
     submit_btn.click(
         fn=run_prediction,