Example caching and space duplication

app.py

@@ -5,6 +5,12 @@ from diffusers import LMSDiscreteScheduler
 from mixdiff import StableDiffusionCanvasPipeline, Text2ImageRegion
 
 article = """
+## Usage
+
+In this demo you can use Mixture of Diffusers to configure a canvas made up of 3 diffusion regions. Play around with the prompts and guidance values in each region! You can also change increment the overlap between regions if seams appear in the image.
+
+In the full version of Mixture of Diffusers you will find further freedom to configure the regions in the canvas. Check the [github repo](https://github.com/albarji/mixture-of-diffusers)!
+
 ## Motivation
 
 Current image generation methods, such as Stable Diffusion, struggle to position objects at specific locations. While the content of the generated image (somewhat) reflects the objects present in the prompt, it is difficult to frame the prompt in a way that creates an specific composition. For instance, take a prompt expressing a complex composition such as
@@ -30,12 +36,6 @@ The method proposed here strives to provide a better tool for image composition
 
 Using several diffusion processes in parallel has also practical advantages when generating very large images, as the GPU memory requirements are similar to that of generating an image of the size of a single tile.
 
-## Usage
-
-In this demo you can use Mixture of Diffusers to configure a canvas made up of 3 diffusion regions. Play around with the prompts and guidance values in each region! You can also change increment the overlap between regions if seams appear in the image.
-
-In the full version of Mixture of Diffusers you will find further freedom to configure the regions in the canvas. Check the [github repo](https://github.com/albarji/mixture-of-diffusers)!
-
 ## Responsible use
 
 The same recommendations as in Stable Diffusion apply, so please check the corresponding [model card](https://huggingface.co/CompVis/stable-diffusion-v1-4).
@@ -44,7 +44,7 @@ More broadly speaking, always bear this in mind: YOU are responsible for the con
 
 ## Gallery
 
-Here are some relevant illustrations I have created using this software (and putting quite a few hours into them!).
+Here are some relevant illustrations created using this software (and putting quite a few hours into them!).
 
 ### Darkness Dawning
 
@@ -94,19 +94,46 @@ def generate(prompt1, prompt2, prompt3, gc1, gc2, gc3, overlap, steps, seed):
     )["sample"][0]
 
 with gr.Blocks(title="Mixture of Diffusers") as demo:
-    gr.Markdown("# Mixture of Diffusers")
+    gr.HTML(
+        """
+            <div style="text-align: center; max-width: 700px; margin: 0 auto;">
+              <div
+                style="
+                  display: inline-flex;
+                  align-items: center;
+                  gap: 0.8rem;
+                  font-size: 1.75rem;
+                "
+              >
+                <h1 style="font-weight: 1000; margin-bottom: 7px; line-height: normal;">
+                  Mixture of Diffusers
+                </h1>
+              </div>
+              <p style="margin-bottom: 10px; font-size: 94%">
+                <a href="https://arxiv.org/abs/2302.02412">[Paper]</a>  <a href="https://github.com/albarji/mixture-of-diffusers">[Code in Github]</a>  <a href="https://huggingface.co/spaces/albarji/mixture-of-diffusers?duplicate=true">
+              </p>
+            </div>
+        """
+    )
+    gr.HTML("""
+        <p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings.
+        <br/>
+        <a href="https://huggingface.co/spaces/albarji/mixture-of-diffusers?duplicate=true">
+        <img style="margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
+        <p/>
+    """)
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Left region")
-            left_prompt = gr.Textbox(lines=2, label="Prompt")
+            left_prompt = gr.Textbox(lines=2, label="Prompt (what do you want to see in the left side of the image?)")
             left_gs = gr.Slider(minimum=0, maximum=15, value=8, step=1, label="Guidance scale")
         with gr.Column(scale=1):
             gr.Markdown("### Center region")
-            center_prompt = gr.Textbox(lines=2, label="Prompt")
+            center_prompt = gr.Textbox(lines=2, label="Prompt (what do you want to see in the center of the image?)")
             center_gs = gr.Slider(minimum=0, maximum=15, value=8, step=1, label="Guidance scale")
         with gr.Column(scale=1):
             gr.Markdown("### Right region")
-            right_prompt = gr.Textbox(lines=2, label="Prompt")
+            right_prompt = gr.Textbox(lines=2, label="Prompt (what do you want to see in the right side of the image?)")
             right_gs = gr.Slider(minimum=0, maximum=15, value=8, step=1, label="Guidance scale")
     gr.Markdown("### General parameters")
     with gr.Row():
@@ -132,11 +159,29 @@ with gr.Blocks(title="Mixture of Diffusers") as demo:
                     50,
                     7178915308
                 ],
+                [
+                    "Abstract decorative illustration, by joan miro and gustav klimt and marlina vera and loish, elegant, intricate, highly detailed, smooth, sharp focus, vibrant colors, artstation, stunning masterpiece",
+                    "Abstract decorative illustration, by joan miro and gustav klimt and marlina vera and loish, elegant, intricate, highly detailed, smooth, sharp focus, vibrant colors, artstation, stunning masterpiece",
+                    "Abstract decorative illustration, by joan miro and gustav klimt and marlina vera and loish, elegant, intricate, highly detailed, smooth, sharp focus, vibrant colors, artstation, stunning masterpiece",
+                    8, 8, 8,
+                    256,
+                    35,
+                    21156517
+                ],
+                [
+                    "Magical diagrams and runes written with chalk on a blackboard, elegant, intricate, highly detailed, smooth, sharp focus, artstation, stunning masterpiece",
+                    "Magical diagrams and runes written with chalk on a blackboard, elegant, intricate, highly detailed, smooth, sharp focus, artstation, stunning masterpiece",
+                    "Magical diagrams and runes written with chalk on a blackboard, elegant, intricate, highly detailed, smooth, sharp focus, artstation, stunning masterpiece",
+                    12, 12, 12,
+                    256,
+                    35,
+                    12591765619
+                ]
             ],
             inputs=[left_prompt, center_prompt, right_prompt, left_gs, center_gs, right_gs, overlap, steps, seed],
-            # outputs=output,
-            # fn=generate,
-            # cache_examples=True
+            outputs=output,
+            fn=generate,
+            cache_examples=True
         )
     
     button.click(