add examples

src/app.py

@@ -114,10 +114,20 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(type="pil", label="Input Image", image_mode="RGBA")
-            run_button = gr.Button(value="Relight Image")
+            input_image = gr.Image(
+                label="Input Image (RGBA)",
+                image_mode="RGBA",
+                type="pil",
+            )
+            run_button = gr.Button(
+                value="Relight Image",
+            )
         with gr.Column():
-            output_image = gr.Image(label="Result")
+            output_image = gr.Image(
+                label="Relighted Image (RGB)",
+                image_mode="RGB",
+                type="pil",
+            )
 
     with gr.Accordion("Advanced Settings", open=True):
         prompt = gr.Textbox(
@@ -221,6 +231,28 @@ with gr.Blocks() as demo:
                 25,
                 69_420,
             ],
+            [
+                "examples/chair.png",
+                "Right",
+                "god rays, fluffy clouds, peaceful surreal atmosphere, high-quality, HEIC, CR2, NEF",
+                "dirty, messy, worst quality, low quality, watermark, signature, jpeg artifacts, deformed, monochrome, black and white",
+                0.9,
+                0.5,
+                1.25,
+                25,
+                69,
+            ],
+            [
+                "examples/bunny.png",
+                "Left",
+                "grass field, high-quality, HEIC, CR2, NEF",
+                "dirty, messy, worst quality, low quality, watermark, signature, jpeg artifacts, deformed, monochrome, black and white",
+                0.9,
+                0.5,
+                1.25,
+                25,
+                420,
+            ],
         ],
         inputs=[
             input_image,
@@ -235,7 +267,8 @@ with gr.Blocks() as demo:
         ],
         outputs=output_image,
         fn=process,
-        cache_examples="lazy",  # type: ignore
+        cache_examples=True,
+        cache_mode="lazy",
         run_on_click=False,
     )
 

src/utils.py

@@ -53,8 +53,7 @@ def resize_modulo_8(
     resample: Image.Resampling | None = None,
     on_short: bool = True,
 ) -> Image.Image:
-    """
-    Resize an image respecting the aspect ratio and ensuring the size is a multiple of 8.
+    """Resize an image respecting the aspect ratio and ensuring the size is a multiple of 8.
 
     The `on_short` parameter determines whether the resizing is based on the shortest side.
     """
@@ -73,8 +72,7 @@ class LightingPreference(str, Enum):
     NONE = auto()
 
     def get_init_image(self, width: int, height: int, interval: tuple[float, float] = (0.0, 1.0)) -> Image.Image | None:
-        """
-        Generate an image with a linear gradient based on the lighting preference.
+        """Generate an image with a linear gradient based on the lighting preference.
 
         In the original code, interval is always (0., 1.) ; we added it as a parameter to make the function more
         flexible and allow for less contrasted images with a smaller interval.