Light Migration

app.py

@@ -201,7 +201,6 @@ with gr.Blocks() as demo:
         gr.Markdown("## 💡 Qwen Image Edit — Light Migration")
         gr.Markdown("""
             Transfer lighting and color tones from a reference image to your source image ✨
-            
             Using [dx8152's Qwen-Edit-2509-Light-Migration LoRA](https://huggingface.co/dx8152/Qwen-Edit-2509-Light-Migration) 
             and [Phr00t/Qwen-Image-Edit-Rapid-AIO](https://huggingface.co/Phr00t/Qwen-Image-Edit-Rapid-AIO/tree/main) for 4-step inference 💨
         """)
@@ -272,7 +271,31 @@ with gr.Blocks() as demo:
             with gr.Column():
                 result = gr.Image(label="Output Image", interactive=False)
                 output_seed = gr.Number(label="Seed Used", interactive=False)
-
+        
+        gr.Examples(
+            examples=[
+                # Character 1 with 3 different lights
+                ["character_1.png", "light_1.png"],
+                ["character_1.png", "light_3.jpeg"],
+                ["character_1.png", "light_5.png"],
+                # Character 2 with 3 different lights
+                ["character_2.png", "light_2.png"],
+                ["character_2.png", "light_4.png"],
+                ["character_2.png", "light_6.png"],
+                # Place 1 with 3 different lights
+                ["place_1.png", "light_1.png"],
+                ["place_1.png", "light_4.png"],
+                ["place_1.png", "light_6.png"],
+            ],
+            inputs=[
+                image, light_source
+            ],
+            outputs=outputs,
+            fn=infer_light_migration,
+            cache_examples=True,
+            cache_mode="lazy",
+            elem_id="examples"
+        )
     inputs = [
         image, light_source, prompt,
         seed, randomize_seed, true_guidance_scale,