fix: ad classes ui

This commit is contained in:
Dowon
2024-03-01 11:34:26 +09:00
parent 9d46fcd714
commit aa9e4553e9

View File

@@ -88,7 +88,8 @@ def on_generate_click(state: dict, *values: Any):
def on_ad_model_update(model: str):
if "-world" in model:
return gr.update(
visible=True, placeholder="Comma separated class names to detect."
visible=True,
placeholder="Comma separated class names to detect, ex: 'person,cat'. default: COCO 80 classes",
)
return gr.update(visible=False, placeholder="")
@@ -169,6 +170,7 @@ def one_ui_group(n: int, is_img2img: bool, webui_info: WebuiInfo):
w = Widgets()
eid = partial(elem_id, n=n, is_img2img=is_img2img)
with gr.Group():
with gr.Row():
model_choices = (
[*webui_info.ad_model_list, "None"]
@@ -185,6 +187,7 @@ def one_ui_group(n: int, is_img2img: bool, webui_info: WebuiInfo):
elem_id=eid("ad_model"),
)
with gr.Row():
w.ad_model_classes = gr.Textbox(
label="ADetailer model classes" + suffix(n),
value="",
@@ -192,13 +195,15 @@ def one_ui_group(n: int, is_img2img: bool, webui_info: WebuiInfo):
elem_id=eid("ad_classes"),
)
w.ad_model_classes.change(
w.ad_model.change(
on_ad_model_update,
inputs=w.ad_model,
outputs=w.ad_model_classes,
queue=False,
)
gr.HTML("<br>")
with gr.Group():
with gr.Row(elem_id=eid("ad_toprow_prompt")):
w.ad_prompt = gr.Textbox(