fixed order

app.py

@@ -12,6 +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)
+
 
 # %% prediction on an image
 
@@ -84,6 +87,9 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             rows=1, height=256, type="pil"
     )
     
+    gta_gallery.select(fn=load_example, inputs=[gta_gallery], outputs=[image_input])
+    city_gallery.select(fn=load_example, inputs=[city_gallery], outputs=[image_input])
+    
     submit_btn.click(
         fn=run_prediction,
         inputs=[image_input, model_selector],
@@ -92,13 +98,6 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
 
     gr.Markdown("Made by group 21 semantic segmentation project. ")
 
-def load_example(example_img):
-    return gr.update(value=example_img)
-
-# On click: update image_input with selected example
-gta_gallery.select(fn=load_example, inputs=[gta_gallery], outputs=[image_input])
-city_gallery.select(fn=load_example, inputs=[city_gallery], outputs=[image_input])
-
 demo.launch()