Update app.py

app.py

@@ -14,7 +14,7 @@ resolution = 512
 # checkpoint of the converted Stable Diffusion from KerasCV
 model_ckpt = "nielsgl/dreambooth-bored-ape"
 pipeline = StableDiffusionPipeline.from_pretrained(model_ckpt)
-pipeline.to("cuda")
+# pipeline.to("cuda")
 
 unique_id = "drawbayc"
 class_label = "monkey"
@@ -106,7 +106,7 @@ with gr.Blocks() as demo:
             output = gr.Gallery(label="Outputs").style(grid=(1,2))
     prompt_btn.click(infer, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=[output])
     with gr.Row():
-        gr.Examples(examples, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=output, fn=infer)
+        gr.Examples(examples, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=output, fn=infer, cache_examples=True)
     with gr.Row():
         with gr.Column():
             gr.Markdown(model_card_2)