update

app.py

@@ -16,6 +16,7 @@ import spaces
 import argparse
 
 import os
+import shutil
 import cv2
 import gradio as gr
 import numpy as np
@@ -36,6 +37,10 @@ args = parser.parse_args()
 
 huggingface_hub.login(os.getenv('HF_TOKEN'))
 
+try:
+    shutil.rmtree('gradio_cached_examples')
+except FileNotFoundError:
+    print("cache folder not exist")
 
 class Generator:
     def __init__(self):
@@ -189,8 +194,8 @@ def create_demo():
         with gr.Row():
             with gr.Column():
                 with gr.Row():
-                    ref_image1 = gr.Image(label="ref image 1", type="filepath", height=256)
-                    ref_image2 = gr.Image(label="ref image 2", type="filepath", height=256)
+                    ref_image1 = gr.Image(label="ref image 1", type="numpy", height=256)
+                    ref_image2 = gr.Image(label="ref image 2", type="numpy", height=256)
                 with gr.Row():
                     ref_task1 = gr.Dropdown(choices=["ip", "id", "style"], value="ip", label="task for ref image 1")
                     ref_task2 = gr.Dropdown(choices=["ip", "id", "style"], value="ip", label="task for ref image 2")
@@ -254,12 +259,6 @@ def create_demo():
                     'a perfume under spotlight',
                     116150031980664704,
                 ],
-            ]
-            gr.Examples(examples=example_inps, inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed], label='IP task', cache_examples='lazy',
-                        outputs=[output_image, debug_image, seed_output],
-                        fn=generate_image,)
-
-            example_inps = [
                 [
                     'example_inputs/hinton.jpeg',
                     None,
@@ -268,17 +267,6 @@ def create_demo():
                     'portrait, Chibi',
                     5443415087540486371,
                 ],
-            ]
-            gr.Examples(
-                examples=example_inps,
-                inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed],
-                label='ID task (similar to PuLID, will only refer to the face)',
-                cache_examples='lazy',
-                outputs=[output_image, debug_image, seed_output],
-                fn=generate_image,
-            )
-
-            example_inps = [
                 [
                     'example_inputs/mickey.png',
                     None,
@@ -295,12 +283,6 @@ def create_demo():
                     'generate a same style image. A pavilion by the river, and the distant mountains are endless',
                     5248066378927500767,
                 ],
-            ]
-            gr.Examples(examples=example_inps, inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed], label='Style task', cache_examples='lazy',
-                        outputs=[output_image, debug_image, seed_output],
-                        fn=generate_image,)
-
-            example_inps = [
                 [
                     'example_inputs/shirt.png',
                     'example_inputs/skirt.jpeg',
@@ -317,17 +299,6 @@ def create_demo():
                     'the woman wearing a dress, In the banquet hall',
                     7698454872441022867,
                 ],
-            ]
-            gr.Examples(
-                examples=example_inps,
-                inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed],
-                label='Try-On task',
-                cache_examples='lazy',
-                outputs=[output_image, debug_image, seed_output],
-                fn=generate_image,
-            )
-
-            example_inps = [
                 [
                     'example_inputs/dog1.png',
                     'example_inputs/dog2.png',
@@ -356,7 +327,7 @@ def create_demo():
             gr.Examples(
                 examples=example_inps,
                 inputs=[ref_image1, ref_image2, ref_task1, ref_task2, prompt, seed],
-                label='Multi IP',
+                label='row 1-4: IP task; row 5: ID task; row 6-7: Style task. row 8-9: Try-On task; row 10-12: Multi IP',
                 cache_examples='lazy',
                 outputs=[output_image, debug_image, seed_output],
                 fn=generate_image,