Update app.py

app.py

@@ -10,6 +10,7 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是此项目只支持GPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
+CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
 MAX_IMAGE_SIZE = 4096
 
 if torch.cuda.is_available():
@@ -63,8 +64,8 @@ def infer(
     return image, seed
 
 examples = [
-    "a cat eating a piece of cheese",
-    "a ROBOT riding a BLUE horse on Mars, photorealistic, 4k",
+    "nahida (genshin impact)",
+    "klee (genshin impact)",
 ]
 
 css = '''
@@ -139,6 +140,14 @@ with gr.Blocks(css=css) as demo:
                 value=28,
             )
 
+    gr.Examples(
+        examples=examples,
+        inputs=prompt,
+        outputs=[result, seed],
+        fn=generate,
+        cache_examples=CACHE_EXAMPLES,
+    )
+    
     use_negative_prompt.change(
         fn=lambda x: gr.update(visible=x),
         inputs=use_negative_prompt,