download examples from Space repo at startup

app.py

@@ -12,6 +12,22 @@ import numpy as np
 # Download checkpoint from HF Hub at startup
 checkpoint_path = hf_hub_download(repo_id="HadiZayer/MagicFixup", filename="magicfu_weights")
 
+# Download example images from the Space repo at startup
+EXAMPLE_PAIRS = [
+    ("examples/fox_drinking_og.png",    "examples/fox_drinking__edit__01.png"),
+    ("examples/palm_tree_og.png",       "examples/palm_tree__edit__01.png"),
+    ("examples/kingfisher_og.png",      "examples/kingfisher__edit__001.png"),
+    ("examples/pipes_og.png",           "examples/pipes__edit__01.png"),
+    ("examples/dog_beach_og.png",       "examples/dog_beach__edit__003.png"),
+]
+examples = [
+    [
+        hf_hub_download(repo_id="HadiZayer/MagicFixup", filename=og,   repo_type="space"),
+        hf_hub_download(repo_id="HadiZayer/MagicFixup", filename=edit, repo_type="space"),
+    ]
+    for og, edit in EXAMPLE_PAIRS
+]
+
 magic_fixup = MagicFixup(model_path=checkpoint_path)
 
 
@@ -40,7 +56,7 @@ demo = gr.Interface(
         gr.Image(type="pil", image_mode="RGBA", label="Coarse Edit (with alpha mask)"),
     ],
     outputs=gr.Image(label="Result"),
-    examples="examples",
+    examples=examples,
     title="Magic Fixup",
     description="Upload your original image and a coarse edit (PNG with alpha channel marking the edited region). Magic Fixup will refine the edit to look photorealistic.",
 )