From a3bc7460b370e9c623368f0ef4bf8f6609a42e0d Mon Sep 17 00:00:00 2001
From: AI Lab <129358391+1038lab@users.noreply.github.com>
Date: Sun, 1 Jun 2025 14:25:45 -0700
Subject: [PATCH] Update AILab_SegmentV2.py

---
 AILab_SegmentV2.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/AILab_SegmentV2.py b/AILab_SegmentV2.py
index 0aa3a24..ad781da 100644
--- a/AILab_SegmentV2.py
+++ b/AILab_SegmentV2.py
@@ -268,10 +268,10 @@ class SegmentV2:
         return (pil2tensor(result_image), mask_tensor, mask_image_vis)
 
 NODE_CLASS_MAPPINGS = {
-    "AILab_SegmentV2": SegmentV2,
+    "SegmentV2": SegmentV2,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "AILab_SegmentV2": "Segmentation V2 (RMBG)",
+    "SegmentV2": "Segmentation V2 (RMBG)",
 }