refactor: rename cloth_category to inpaint_mask_cloth_category

This commit is contained in:
Manuel Schmid 2024-01-25 10:45:33 +01:00
parent d1c55db2bc
commit d2b697f33b
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 7 additions and 7 deletions

View File

@ -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():