Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-30 19:21:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: ad classes ui

Browse Source
This commit is contained in:
Dowon
2024-03-01 11:34:26 +09:00
parent 9d46fcd714
commit aa9e4553e9
1 changed files with 32 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
adetailer/ui.py
View File

@@ -88,7 +88,8 @@ def on_generate_click(state: dict, *values: Any):
def on_ad_model_update(model: str): def on_ad_model_update(model: str):
if "-world" in model: if "-world" in model:
return gr.update( 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="") return gr.update(visible=False, placeholder="")
@@ -169,35 +170,39 @@ def one_ui_group(n: int, is_img2img: bool, webui_info: WebuiInfo):
w = Widgets() w = Widgets()
eid = partial(elem_id, n=n, is_img2img=is_img2img) eid = partial(elem_id, n=n, is_img2img=is_img2img)
with gr.Row(): with gr.Group():
model_choices = ( with gr.Row():
[*webui_info.ad_model_list, "None"] model_choices = (
if n == 0 [*webui_info.ad_model_list, "None"]
else ["None", *webui_info.ad_model_list] if n == 0
) else ["None", *webui_info.ad_model_list]
)
w.ad_model = gr.Dropdown( w.ad_model = gr.Dropdown(
label="ADetailer model" + suffix(n), label="ADetailer model" + suffix(n),
choices=model_choices, choices=model_choices,
value=model_choices[0], value=model_choices[0],
visible=True, visible=True,
type="value", type="value",
elem_id=eid("ad_model"), elem_id=eid("ad_model"),
) )
w.ad_model_classes = gr.Textbox( with gr.Row():
label="ADetailer model classes" + suffix(n), w.ad_model_classes = gr.Textbox(
value="", label="ADetailer model classes" + suffix(n),
visible=False, value="",
elem_id=eid("ad_classes"), visible=False,
) elem_id=eid("ad_classes"),
)
w.ad_model_classes.change( w.ad_model.change(
on_ad_model_update, on_ad_model_update,
inputs=w.ad_model, inputs=w.ad_model,
outputs=w.ad_model_classes, outputs=w.ad_model_classes,
queue=False, queue=False,
) )
gr.HTML("<br>")
with gr.Group(): with gr.Group():
with gr.Row(elem_id=eid("ad_toprow_prompt")): with gr.Row(elem_id=eid("ad_toprow_prompt")):