diff --git a/comfy_extras/nodes_bg_removal.py b/comfy_extras/nodes_bg_removal.py index 91de3b27b..b88d19aba 100644 --- a/comfy_extras/nodes_bg_removal.py +++ b/comfy_extras/nodes_bg_removal.py @@ -12,7 +12,7 @@ class LoadBackgroundRemovalModel(IO.ComfyNode): node_id="LoadBackgroundRemovalModel", category="loaders", inputs=[ - IO.Combo.Input("bg_removal_name", options=sorted(files)), + IO.Combo.Input("bg_removal_name", options=sorted(files), tooltip="The model used to remove backgrounds from images"), ], outputs=[ IO.BackgroundRemoval.Output("bg_model") @@ -34,11 +34,11 @@ class RemoveBackground(IO.ComfyNode): display_name="Remove Background", category="image/background removal", inputs=[ - IO.Image.Input("image"), - IO.BackgroundRemoval.Input("bg_removal_model") + IO.Image.Input("image", tooltip="Input image to remove the background from"), + IO.BackgroundRemoval.Input("bg_removal_model", tooltip="Background removal model used to generate the mask") ], outputs=[ - IO.Mask.Output("mask") + IO.Mask.Output("mask", tooltip="Generated foreground mask") ] ) @classmethod