__init__.py

"""
SETU Translation Model for Hugging Face Transformers

This package provides the SETU (Script-agnostic English Translation Unifier) model
for translating multiscript, multilingual, and informal text into clean, formal English.

Usage:
    from transformers import AutoModel
    
    # Load the model
    model = AutoModel.from_pretrained("santoshdahal/setu", trust_remote_code=True)
    
    # Translate text
    result = model("mero name santosh ho")
    print("Translation:", result)
"""

from transformers import AutoConfig, AutoModel
try:
    from .configuration_setu_translation import SetuTranslationConfig
    from .modeling_setu_translation import SetuTranslationModel
except ImportError:
    from configuration_setu_translation import SetuTranslationConfig
    from modeling_setu_translation import SetuTranslationModel

# Register the model configuration and model class
AutoConfig.register("setu_translation", SetuTranslationConfig)
AutoModel.register(SetuTranslationConfig, SetuTranslationModel)

__all__ = [
    "SetuTranslationConfig",
    "SetuTranslationModel",
]