Add examples

edit_app.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import math
+import os
 import random
 
 import gradio as gr
@@ -113,6 +114,10 @@ def reset():
     return [0, "Randomize Seed", 1371, "Fix CFG", 7.5, 1.5, None]
 
 
+def process_example(input_image: Image.Image, instruction: str, seed: int) -> Image.Image:
+    return generate(input_image, instruction, 50, False, seed, False, 7.5, 1.5)[-1]
+
+
 with gr.Blocks() as demo:
     gr.HTML(
         """<h1 style="font-weight: 900; margin-bottom: 7px;">
@@ -160,6 +165,17 @@ InstructPix2Pix: Learning to Follow Image Editing Instructions
         text_cfg_scale = gr.Number(value=7.5, label="Text CFG", interactive=True)
         image_cfg_scale = gr.Number(value=1.5, label="Image CFG", interactive=True)
 
+    gr.Examples(
+        examples=[
+            ["imgs/example.jpg", "Turn him into a cyborg", 0],
+            ["imgs/example.jpg", "Have him smile", 0],
+        ],
+        inputs=[input_image, instruction, seed],
+        outputs=edited_image,
+        fn=process_example,
+        cache_examples=os.getenv("CACHE_EXAMPLES") == "1",
+    )
+
     gr.Markdown(help_text)
 
     load_button.click(