more updates

app.py

@@ -20,6 +20,7 @@ import utils
 
 dtype = torch.float16
 device = torch.device("cuda")
+examples = [[True, True, "SD T2I", 4], [False, True, "Würstchen (T2I)", 4]]
 
 pipeline_mapping = {
     "SD T2I": (DiffusionPipeline, "runwayml/stable-diffusion-v1-5"),
@@ -202,7 +203,7 @@ def generate(
         print("Avg per step", (end_time - start_time) / num_inference_steps)
 
 
-with gr.Blocks() as demo:
+with gr.Blocks(css="style.css") as demo:
     do_torch_compile = gr.Checkbox(label="Enable torch.compile()?")
     use_channels_last = gr.Checkbox(label="Use `channels_last` memory layout?")
     pipeline_to_benchmark = gr.Dropdown(
@@ -218,6 +219,14 @@ with gr.Blocks() as demo:
         step=1,
         value=1,
     )
+    gr.Examples(
+        examples=examples,
+        inputs=[pipeline_to_benchmark, batch_size, use_channels_last, do_torch_compile],
+        outputs="text",
+        fn=generate,
+        cache_examples=True,
+    )
+
     btn = gr.Button("Benchmark!").style(
         margin=False,
         rounded=(False, True, True, False),
@@ -227,6 +236,7 @@ with gr.Blocks() as demo:
     btn.click(
         fn=generate,
         inputs=[pipeline_to_benchmark, batch_size, use_channels_last, do_torch_compile],
+        outputs="text"
     )
 
-demo.launch()
+demo.launch(show_error=True)

style.css

@@ -0,0 +1,16 @@
+h1 {
+    text-align: center;
+  }
+  
+  #duplicate-button {
+    margin: auto;
+    color: #fff;
+    background: #1565c0;
+    border-radius: 100vh;
+  }
+  
+  #component-0 {
+    max-width: 730px;
+    margin: auto;
+    padding-top: 1.5rem;
+  }