examples

app.py

@@ -20,6 +20,21 @@ MAX_SEED = np.iinfo(np.int32).max
 IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
+EXAMPLES = [
+    [
+        {
+            "background": "https://media.roboflow.com/spaces/doge-2-image.jpeg",
+            "layers": ["https://media.roboflow.com/spaces/doge-2-mask.png"],
+            "composite": None
+        },
+        "spider tattoo",
+        42,
+        False,
+        0.9,
+        30
+    ]
+]
+
 pipe = FluxInpaintPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
 
@@ -157,6 +172,25 @@ with gr.Blocks() as demo:
             with gr.Accordion("Debug", open=False):
                 output_mask_component = gr.Image(
                     type='pil', image_mode='RGB', label='Input mask')
+    with gr.Row():
+        gr.Examples(
+            fn=process,
+            examples=EXAMPLES,
+            inputs=[
+                input_image_editor_component,
+                input_text_component,
+                seed_slicer_component,
+                randomize_seed_checkbox_component,
+                strength_slider_component,
+                num_inference_steps_slider_component
+            ],
+            outputs=[
+                output_image_component,
+                output_mask_component
+            ],
+            run_on_click=True,
+            cache_examples=True
+        )
 
     submit_button_component.click(
         fn=process,