Update app.py

app.py

@@ -28,7 +28,7 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, times
 
 # Inference function.
 @spaces.GPU(enable_queue=True)
-def generate_image(prompt, option, progress=gr.Progress()):
+def generate(prompt, option, progress=gr.Progress()):
     global step_loaded
     print(prompt, option)
     ckpt, step = opts[option]
@@ -69,14 +69,25 @@ with gr.Blocks(css="style.css") as demo:
     img = gr.Image(label="SDXL-Lighting Generated Image")
 
     prompt.submit(
-        fn=generate_image,
+        fn=generate,
         inputs=[prompt, option],
         outputs=img,
     )
     submit.click(
-        fn=generate_image,
+        fn=generate,
         inputs=[prompt, option],
         outputs=img,
     )
+
+    gr.Examples(
+        fn=generate,
+        examples=[
+            ["A girl smiling", "4 Steps"],
+            ["An astronaut riding a horse", "4 Steps"]
+        ],
+        inputs=[prompt, option],
+        outputs=img,
+        cache_examples=True,
+    )
     
 demo.queue().launch()