test examples

app.py

@@ -4,6 +4,7 @@ from typing import Tuple, Optional
 
 import gradio as gr
 import numpy as np
+import requests
 import spaces
 import torch
 from PIL import Image, ImageFilter
@@ -23,6 +24,43 @@ IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 PIPE = FluxInpaintPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
+CLIENT = Client("SkalskiP/florence-sam-masking")
+
+
+EXAMPLES = [
+    [
+        None,
+        {
+            "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-image.png", stream=True).raw),
+            "layers": [Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-mask-2-removebg.png", stream=True).raw)],
+            "composite": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-composite-2.png", stream=True).raw),
+        },
+        "little lion",
+        None,
+        5,
+        5,
+        42,
+        False,
+        0.85,
+        30
+    ],
+    [
+        None,
+        {
+            "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-image.png", stream=True).raw),
+            "layers": [Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-mask-3-removebg.png", stream=True).raw)],
+            "composite": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-composite-3.png", stream=True).raw),
+        },
+        "tribal tattoos",
+        None,
+        5,
+        5,
+        42,
+        False,
+        0.85,
+        30
+    ]
+]
 
 
 def calculate_image_dimensions_for_flux(
@@ -84,7 +122,7 @@ def set_client_for_session(request: gr.Request):
     return Client("SkalskiP/florence-sam-masking", headers={"X-IP-Token": x_ip_token})
 
 
-@spaces.GPU(duration=100)
+@spaces.GPU(duration=50)
 def run_flux(
     image: Image.Image,
     mask: Image.Image,
@@ -187,7 +225,7 @@ with gr.Blocks() as demo:
 
             with gr.Row():
                 inpainting_prompt_text_component = gr.Text(
-                    label="Prompt",
+                    label="Inpainting prompt",
                     show_label=False,
                     max_lines=1,
                     placeholder="Enter text to generate inpainting",
@@ -198,7 +236,7 @@ with gr.Blocks() as demo:
 
             with gr.Accordion("Advanced Settings", open=False):
                 masking_prompt_text_component = gr.Text(
-                    label="Prompt",
+                    label="Masking prompt",
                     show_label=False,
                     max_lines=1,
                     placeholder="Enter text to generate masking",
@@ -264,6 +302,28 @@ with gr.Blocks() as demo:
             with gr.Accordion("Debug", open=False):
                 output_mask_component = gr.Image(
                     type='pil', image_mode='RGB', label='Input mask', format="png")
+    gr.Examples(
+        fn=process,
+        examples=EXAMPLES,
+        inputs=[
+            client_component,
+            input_image_editor_component,
+            inpainting_prompt_text_component,
+            masking_prompt_text_component,
+            mask_inflation_slider_component,
+            mask_blur_slider_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,