Update app.py

app.py

@@ -49,6 +49,16 @@ def generate_image(prompt, ckpt):
     return results.images[0]
 
 
+examples = [
+    "a cat eating a piece of cheese",
+    "a ROBOT riding a BLUE horse on Mars, photorealistic",
+    "Ironman VS Hulk, ultrarealistic",
+    "a CUTE robot artist painting on an easel",
+    "Astronaut in a jungle, cold color palette, oil pastel, detailed, 8k",
+    "An alien holding sign board contain word 'Flash', futuristic, neonpunk",
+    "Kids going to school, Anime style"
+]
+
 
 # Gradio Interface
 
@@ -60,7 +70,14 @@ with gr.Blocks(css=CSS) as demo:
             prompt = gr.Textbox(label='Enter your prompt (English)', scale=8)
             ckpt = gr.Dropdown(label='Select inference steps',choices=['1-Step', '4-Step'], value='4-Step', interactive=True)
             submit = gr.Button(scale=1, variant='primary')
-    img = gr.Image(label='DMD2 Generated Image')
+    img = gr.Image(label='DMD2 Generated Image')    
+    gr.Examples(
+        examples=examples,
+        inputs=prompt,
+        outputs=img,
+        fn=generate_image,
+        cache_examples=True,
+    )
 
     prompt.submit(fn=generate_image,
                  inputs=[prompt, ckpt],