From 5ea07f07e68e37d0845a6dac92c193b2c25d3187 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Mon, 31 Jul 2023 11:24:14 +0900
Subject: [PATCH] [fix] misspelling

---
 __init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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