diff --git a/__init__.py b/__init__.py index 6f7ebc9..4947104 100644 --- a/__init__.py +++ b/__init__.py @@ -48,10 +48,10 @@ class RemoveImageBackgroundARB: } RETURN_TYPES = ("IMAGE",) - FUNCTION = "arb_remover" + FUNCTION = "agb_remover" CATEGORY = "image" - def arb_remover(self, image:torch.Tensor): + def agb_remover(self, image:torch.Tensor): npa = image2nparray(image) print(npa.ndim) rmb = rmbg_fn(npa) @@ -76,5 +76,5 @@ def nparray2image(narray:np.array): return tensor NODE_CLASS_MAPPINGS = { - "Remove Image Background (ARB)": RemoveImageBackgroundARB + "Remove Image Background (AGB)": RemoveImageBackgroundARB } \ No newline at end of file