Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 import gradio as gr
 from PIL import Image
 import spaces
+import tqdm
 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
 
 device = "cuda"
@@ -40,9 +41,13 @@ footer {
 """
 
 @spaces.GPU
-def gen(prompt, negative, width, height):
+def gen(prompt, negative, width, height, use_add, progress=gr.Progress()):
+    if use_add:
+        text = f"{prompt}, {prompt_add}"
+    else:
+        text = f"{prompt"
     prior_output = prior(
-        prompt=f"{prompt}, {prompt_add}",
+        prompt=text,
         height=height,
         width=width,
         negative_prompt=negative,
@@ -52,7 +57,7 @@ def gen(prompt, negative, width, height):
     )
     decoder_output = decoder(
         image_embeddings=prior_output.image_embeddings.half(),
-        prompt=f"{prompt}, {prompt_add}",
+        prompt=text,
         negative_prompt=negative,
         guidance_scale=0.0,
         output_type="pil",
@@ -71,9 +76,11 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             width = gr.Slider(label="Width", minimum=1024, maximum=2048, step=8, value=1024, interactive=True)
             height = gr.Slider(label="Height", minimum=1024, maximum=2048, step=8, value=1024, interactive=True)
+        with gr.Row():
+            use_add = gr.Checkbox(label="Use prompt addition", value=True, interactive=True)
     with gr.Row():
         gallery = gr.Gallery(show_label=False, rows=1, columns=1, allow_preview=True, preview=True)
 
-    button.click(gen, inputs=[prompt, negative, width, height], outputs=gallery)
+    button.click(gen, inputs=[prompt, negative, width, height, use_add], outputs=gallery)
 
-demo.launch(show_api=False)
+demo.queue().launch(show_api=False)