Update app.py

app.py

@@ -252,8 +252,22 @@ output_image_component = gr.Image(type="pil", label="Segmented Image")
 # ]
 # Filter out non-existent example files to prevent errors
 # example_paths = [ex for ex in example_paths if os.path.exists(ex)]
-example_paths = ["cats-on-rock-1948.jpg", "dogs.png", "000000484893.jpg", "https://reachomk.github.io/gen2seg/images/comparison/vertical/7.png", "https://reachomk.github.io/gen2seg/images/comparison/horizontal/11.png", "https://reachomk.github.io/gen2seg/images/comparison/vertical/2.jpg"] # Add paths to example images here if you have them
-
+# Base list of example image paths/URLs
+base_example_images = [
+    "cats-on-rock-1948.jpg",
+    "dogs.png",
+    "000000484893.jpg",
+    "https://reachomk.github.io/gen2seg/images/comparison/vertical/7.png",
+    "https://reachomk.github.io/gen2seg/images/comparison/horizontal/11.png",
+    "https://reachomk.github.io/gen2seg/images/comparison/vertical/2.jpg"
+]
+
+# Generate examples for each image with both model choices
+model_choices_for_examples = list(MODEL_IDS.keys()) # ["SD", "MAE-H"]
+formatted_examples = []
+for img_path_or_url in base_example_images:
+    for model_choice in model_choices_for_examples:
+        formatted_examples.append([img_path_or_url, model_choice])
 iface = gr.Interface(
     fn=segment_image,
     inputs=[input_image_component, model_choice_component],
@@ -261,7 +275,7 @@ iface = gr.Interface(
     title=title,
     description=description,
     article=article,
-    examples=example_paths if example_paths else None, # Pass None if no examples
+    examples=formatted_examples if formatted_examples else None, # Pass None if no examples
     allow_flagging="never",
     theme="shivi/calm_seafoam"
 )