test examples

app.py

@@ -1,3 +1,5 @@
+from functools import partial
+
 import cv2
 import random
 from typing import Tuple, Optional
@@ -29,14 +31,13 @@ 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,
@@ -45,15 +46,14 @@ EXAMPLES = [
         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),
+            "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-5.jpeg", stream=True).raw),
+            "layers": None,
+            "composite": None
         },
-        "tribal tattoos",
-        None,
-        5,
+        "big blue eyes",
+        "eyes",
+        10,
         5,
         42,
         False,
@@ -118,8 +118,11 @@ def process_mask(
 
 
 def set_client_for_session(request: gr.Request):
-    x_ip_token = request.headers['x-ip-token']
-    return Client("SkalskiP/florence-sam-masking", headers={"X-IP-Token": x_ip_token})
+    try:
+        x_ip_token = request.headers['x-ip-token']
+        return Client("SkalskiP/florence-sam-masking", headers={"X-IP-Token": x_ip_token})
+    except:
+        return CLIENT
 
 
 @spaces.GPU(duration=50)
@@ -163,7 +166,7 @@ def process(
     num_inference_steps_slider: int
 ):
     if not inpainting_prompt_text:
-        gr.Info("Please enter a text prompt.")
+        gr.Info("Please enter inpainting text prompt.")
         return None, None
 
     image_path = input_image_editor['background']
@@ -210,6 +213,9 @@ def process(
     return image, mask
 
 
+process_example = partial(process, client=CLIENT)
+
+
 with gr.Blocks() as demo:
     client_component = gr.State()
     gr.Markdown(MARKDOWN)
@@ -303,10 +309,9 @@ with gr.Blocks() as demo:
                 output_mask_component = gr.Image(
                     type='pil', image_mode='RGB', label='Input mask', format="png")
     gr.Examples(
-        fn=process,
+        fn=process_example,
         examples=EXAMPLES,
         inputs=[
-            client_component,
             input_image_editor_component,
             inpainting_prompt_text_component,
             masking_prompt_text_component,
@@ -322,7 +327,7 @@ with gr.Blocks() as demo:
             output_mask_component
         ],
         run_on_click=True,
-        # cache_examples=True
+        cache_examples="lazy"
     )
 
     submit_button_component.click(