Update app.py

app.py

@@ -4,6 +4,9 @@ from diffusers import DiffusionPipeline, UNet2DConditionModel, LCMScheduler
 from huggingface_hub import hf_hub_download
 import spaces
 from PIL import Image
+import os
+
+CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
 
 # Constants
 base = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -62,7 +65,7 @@ examples = [
 
 # Gradio Interface
 
-with gr.Blocks(css=CSS) as demo:
+with gr.Blocks(css=CSS, theme="soft") as demo:
     gr.HTML("<h1><center>Adobe DMD2🦖</center></h1>")
     gr.HTML("<p><center><a href='https://huggingface.co/tianweiy/DMD2'>DMD2</a> text-to-image generation</center></p>")
     with gr.Group():
@@ -76,7 +79,7 @@ with gr.Blocks(css=CSS) as demo:
         inputs=prompt,
         outputs=img,
         fn=generate_image,
-        cache_examples=True,
+        cache_examples=CACHE_EXAMPLES,
     )
 
     prompt.submit(fn=generate_image,