From d2b697f33b55ab5be5eeff5d8a077b64b9dd9478 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Thu, 25 Jan 2024 10:45:33 +0100 Subject: [PATCH] refactor: rename cloth_category to inpaint_mask_cloth_category --- webui.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/webui.py b/webui.py index 2004e0dc..43116f03 100644 --- a/webui.py +++ b/webui.py @@ -206,7 +206,7 @@ with shared.gradio_root: choices=flags.inpaint_mask_models, value=modules.config.default_inpaint_mask_model, visible=False) - cloth_category = gr.Dropdown(label='Cloth category', + inpaint_mask_cloth_category = gr.Dropdown(label='Cloth category', choices=flags.inpaint_mask_cloth_category, value=modules.config.default_inpaint_mask_cloth_category, visible=False) @@ -215,20 +215,20 @@ with shared.gradio_root: def generate_mask(image, mask_model, cloth_category): from extras.inpaint_mask import generate_mask_from_image - return generate_mask_from_image(image, mask_model, extras={"cloth_category": cloth_category}) + return generate_mask_from_image(image, mask_model, {"cloth_category": cloth_category}) generate_mask_button.click(fn=generate_mask, inputs=[ inpaint_input_image, inpaint_mask_model, - cloth_category + inpaint_mask_cloth_category ], outputs=inpaint_mask_image) - inpaint_mask_model.change(lambda x: gr.update(visible=x == 'u2net_cloth_seg'), - inputs=inpaint_mask_model, - outputs=cloth_category, - queue=False, show_progress=False) + inpaint_mask_model.change(lambda x: gr.update(visible=x == 'u2net_cloth_seg'), + inputs=inpaint_mask_model, + outputs=inpaint_mask_cloth_category, + queue=False, show_progress=False) with gr.TabItem(label='Describe') as desc_tab: with gr.Row(): with gr.Column():