Update README

app.py

@@ -8,6 +8,7 @@ from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionS
 from ip_adapter import EasyRef
 from huggingface_hub import hf_hub_download
 import gradio as gr
+import os
 import cv2
 import pillow_avif
 
@@ -25,6 +26,29 @@ def adaptive_resize(w, h, size=1024):
 def res2string(w, h):
     return str(w)+"x"+str(h)
 
+def get_image_path_list(folder_name):
+    image_basename_list = os.listdir(folder_name)
+    image_path_list = sorted([os.path.join(folder_name, basename) for basename in image_basename_list])
+    return image_path_list
+
+def get_example():
+    case = [
+        [
+            get_image_path_list('./assets/aragaki_identity'),
+            "An oil painting of a smiling woman.",
+            "A collage of images, monochrome, lowres, bad anatomy, worst quality, low quality",
+        ],
+        [
+            get_image_path_list('./assets/blindbox_style'),
+            "Donald Trump",
+            "A collage of images, monochrome, lowres, bad anatomy, worst quality, low quality",
+        ],
+    ]
+    return case
+
+def upload_example_to_gallery(images, prompt, negative_prompt):
+    return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False)
+
 base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
 multimodal_llm_path = "Qwen/Qwen2-VL-2B-Instruct"
 ip_ckpt = hf_hub_download(repo_id="zongzhuofan/EasyRef", filename="pytorch_model.bin", repo_type="model")
@@ -116,31 +140,13 @@ with gr.Blocks(css=css) as demo:
         submit.click(fn=generate_image,
                     inputs=[files,prompt,negative_prompt,height, width, scale, num_inference_steps, seed],
                     outputs=gallery)
-        examples = gr.Examples(
-            examples=[
-                [
-                    [
-                        "assets/aragaki_identity/1.jpg",
-                        "assets/aragaki_identity/2.webp",
-                        "assets/aragaki_identity/3.webp",
-                        "assets/aragaki_identity/4.jpeg",
-                        "assets/aragaki_identity/5.webp",
-                    ],
-                    "An oil painting of a smiling woman.",
-                ],
-                [
-                    [
-                        "assets/blindbox_style/1.jpg",
-                        "assets/blindbox_style/2.jpg",
-                        "assets/blindbox_style/3.jpg",
-                        "assets/blindbox_style/4.jpg",
-                        "assets/blindbox_style/5.jpg",
-                    ],
-                    "Donald Trump"
-                ],
-            ],
-            inputs=[files, prompt],
-        )
+    gr.Examples(
+        examples=get_example(),
+        inputs=[files, prompt, negative_prompt],
+        run_on_click=True,
+        fn=upload_example_to_gallery,
+        outputs=[uploaded_files, clear_button, files],
+    )
             
     gr.Markdown("We release our checkpoints for research purposes only. Users are granted the freedom to create images using this tool, but they are expected to comply with local laws and utilize it in a responsible manner. The developers do not assume any responsibility for potential misuse by users.")