setu / __init__.py
santoshdahal's picture
Upload folder using huggingface_hub
785f55b verified
"""
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",
]