Display effection region mask upload regardless of input mode (#2873)

pull/2874/head
Chenlei Hu 2024-05-10 13:43:05 -04:00 committed by GitHub
parent 8d1389da5e
commit aa0a72afcf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 11 deletions

View File

@ -339,17 +339,6 @@ class ControlNetUiGroup(object):
elem_classes=["cnet-close-preview"],
)
with gr.Group(
visible=False, elem_classes=["cnet-mask-image-group"]
) as self.mask_image_group:
self.effective_region_mask = gr.Image(
value=None,
label="Effective Region Mask",
elem_id=f"{elem_id_tabname}_{tabname}_mask_image",
elem_classes=["cnet-effective-region-mask-image"],
interactive=True,
)
with gr.Tab(label="Batch") as self.batch_tab:
self.batch_image_dir = gr.Textbox(
label="Input Directory",
@ -369,6 +358,18 @@ class ControlNetUiGroup(object):
)
self.merge_clear_button = gr.Button("Clear Images")
# Note: effective region mask works with all 3 input types.
with gr.Group(
visible=False, elem_classes=["cnet-mask-image-group"]
) as self.mask_image_group:
self.effective_region_mask = gr.Image(
value=None,
label="Effective Region Mask",
elem_id=f"{elem_id_tabname}_{tabname}_mask_image",
elem_classes=["cnet-effective-region-mask-image"],
interactive=True,
)
if self.photopea:
self.photopea.attach_photopea_output(self.generated_image)