Update app.py

app.py

@@ -77,8 +77,8 @@ class GradioDemo:
             gr.Markdown("# One More Step Demo")
             with gr.Row():
                 with gr.Column():
-                    prompt = gr.Textbox(label="Prompt", value="a cat")
-                    oms_prompt = gr.Textbox(label="OMS Prompt", value="orange cat")
+                    prompt = gr.Textbox(label="Prompt", value="a cat against black ground, studio")
+                    oms_prompt = gr.Textbox(label="OMS Prompt", value="an orange cat")
                     oms_guidance_scale = gr.Slider(label="OMS Guidance Scale", minimum=1.0, maximum=5.0, value=1.5, step=0.1)
                     run_button = gr.Button(value="Generate images")
                     with gr.Accordion("Advanced options", open=False):
@@ -89,7 +89,7 @@ class GradioDemo:
                     output_raw = gr.Image(label="SDXL w/ LCM-LoRA w/o OMS ")
                     output_oms = gr.Image(label="w/ OMS w/o OMS CFG")
                     with gr.Column(visible=False) as oms_cfg_wd:
-                        output_oms_cfg = gr.Image(label=f"w/ OMS w/ OMS CFG")
+                        output_oms_cfg = gr.Image(label="w/ OMS w/ OMS CFG")
 
             ips = [prompt, oms_prompt, oms_guidance_scale, num_steps, sd_guidance_scale, seed]
             run_button.click(fn=self._inference, inputs=ips, outputs=[output_raw, output_oms, output_oms_cfg, oms_cfg_wd])