diff --git a/webui.py b/webui.py index 7ac508e1..3964039a 100644 --- a/webui.py +++ b/webui.py @@ -7,7 +7,7 @@ from modules.cv2win32 import close_all_preview def generate_clicked(prompt, negative_prompt, style_selction, performance_selction, - aspect_ratios_selction, image_number, image_seed): + aspect_ratios_selction, image_number, image_seed, progress=gr.Progress()): p_txt, n_txt = apply_style(style_selction, prompt, negative_prompt) @@ -25,7 +25,7 @@ def generate_clicked(prompt, negative_prompt, style_selction, performance_selcti if not isinstance(seed, int) or seed < 0 or seed > 65535: seed = random.randint(1, 65535) - for i in range(image_number): + for i in progress.tqdm(range(image_number)): imgs = process(p_txt, n_txt, steps, switch, width, height, seed) seed += 1 results += imgs @@ -34,7 +34,7 @@ def generate_clicked(prompt, negative_prompt, style_selction, performance_selcti return results -block = gr.Blocks() +block = gr.Blocks().queue() with block: with gr.Row(): with gr.Column():