Update app.py

app.py

@@ -26,6 +26,17 @@ from modules.model_edit import Step1XParams, Step1XEdit
 
 print("TORCH_CUDA", torch.cuda.is_available())
 
+examples = [
+["examples 2/meme.jpg", "turn into an illustration in studio ghibli style",("examples 2/meme.jpg","examples 2/ghibli_meme.jpg"),],
+["examples 2/celeb_meme.jpg", "replace the gray blazer with a leather jacket",("examples 2/celeb_meme.jpg","examples 2/leather.jpg")],
+["examples 2/cookie.png", "remove the cookie",("examples 2/cookie.png","examples 2/no_cookie.png")],
+["examples 2/poster_orig.jpg", "replace 'lambs' with 'llamas'",("examples 2/poster_orig.jpg","examples 2/poster.jpg")],
+]
+
+def generate_examples(init_image, output_image, prompt):
+    return init_image, output_image, prompt
+
+
 def load_state_dict(model, ckpt_path, device="cuda", strict=False, assign=True):
     if Path(ckpt_path).suffix == ".safetensors":
         state_dict = load_file(ckpt_path, device)
@@ -444,7 +455,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
-                label="编辑指令",
+                label="编辑指令 prompt",
                 value='Remove the person from the image.',
             )
             init_image = gr.Image(label="Input Image", type='pil')
@@ -470,4 +481,12 @@ with gr.Blocks() as demo:
         outputs=[output_image, output_random_seed],
     )
 
+    gr.Examples(
+        examples,
+        inputs=[init_image, prompt, output_image],
+        outputs=[init_image, prompt, output_image],
+        fn=generate_examples,
+        cache_examples=False
+        )
+
 demo.launch()