from .gfrbmg2 import GFrbmg2
from .gfrbmg2plus import GFrbmg2Plus

NODE_CLASS_MAPPINGS = {
    "GFrbmg2": GFrbmg2,
    "GFrbmg2Plus": GFrbmg2Plus
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "GFrbmg2": "🐵 GF Remove Background 2.0",
    "GFrbmg2Plus": "🐵 GF Remove Background (GFrbmg2Plus)"
}