"""
ISNet Background Remover Model
This package provides the ISNet model for background removal using Transformers.
"""

from .modeling_isnet import ISNetConfig, ISNetForImageSegmentation

# Register the model with transformers
from transformers.models.auto.modeling_auto import MODEL_FOR_IMAGE_SEGMENTATION_MAPPING
MODEL_FOR_IMAGE_SEGMENTATION_MAPPING.register(ISNetConfig, ISNetForImageSegmentation)

__all__ = ["ISNetConfig", "ISNetForImageSegmentation"]