Update app.py

app.py

@@ -10,40 +10,48 @@ from diffusers import StableDiffusionSAGPipeline
 
 
 help_text = """
-Self-Attention Guidance (SAG) - Condition-Agnostic Diffusion Guidance Using the Internal Self-Attention
 """
 
 
+
 examples = [
     [
         ' ',
         50,
         "Fix Seed",
-        123,
-        7.5,
+        35934,
+        5.0,
         1.0,
     ],
     [
         '.',
         50,
         "Fix Seed",
-        456,
-        7.5,
+        24865,
+        5.0,
+        1.0,
+    ],
+    [
+        'A poster',
+        50,
+        "Fix Seed",
+        37956,
+        5.0,
         1.0,
     ],
     [
-        'A cute Scottish Fold playing with a ball',
+        'A high-quality living room',
         50,
         "Fix Seed",
-        789,
+        78710,
         5.0,
         1.0,
     ],
     [
-        'A person with a happy dog',
+        'A Scottish Fold playing with a ball',
         50,
         "Fix Seed",
-        901,
+        11511,
         5.0,
         1.0,
     ],
@@ -69,14 +77,18 @@ def main():
         ori_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=0.0).images[0]
         generator = torch.manual_seed(seed)
         sag_image = pipe(prompt, generator=generator, guidance_scale=cfg_scale, sag_scale=sag_scale).images[0]
-        return [ori_image, sag_image]
+        return [ori_image, sag_image, seed]
 
     def reset():
-        return [0, "Randomize Seed", 1371, 5.0, 0.75, None, None]
+        return [0, "Randomize Seed", 42, 5.0, 0.75, None, None]
 
     with gr.Blocks() as demo:
-        gr.HTML("""<h1 style="font-weight: 900; margin-bottom: 7px;">
-           Self-Attention Guidance (SAG) Demo
+        gr.HTML("""<h1 style="font-weight: 900; margin-bottom: 10px;">
+            Self-Attention Guidance (SAG) Demo
+            </h1>
+            <h3 style="font-weight: 500; margin-bottom: 10px;">
+            Condition-Agnostic Diffusion Guidance Using the Internal Self-Attention
+            </h3>
         """)
         with gr.Row():
             with gr.Column(scale=5):
@@ -100,7 +112,7 @@ def main():
             
         with gr.Row():
             cfg_scale = gr.Slider(
-                label="Guidance Scale", minimum=0, maximum=10, value=5.0, step=0.1
+                label="Text Guidance Scale", minimum=0, maximum=10, value=5.0, step=0.1
             )
             sag_scale = gr.Slider(
                 label="Self-Attention Guidance Scale", minimum=0, maximum=1.0, value=0.75, step=0.05
@@ -124,7 +136,7 @@ def main():
                 cfg_scale,
                 sag_scale,
             ],
-            outputs=[ori_image, sag_image],
+            outputs=[ori_image, sag_image, seed],
             cache_examples=False,
         )
 
@@ -140,7 +152,7 @@ def main():
                 cfg_scale,
                 sag_scale,
             ],
-            outputs=[ori_image, sag_image],
+            outputs=[ori_image, sag_image, seed],
         )
         reset_button.click(
             fn=reset,