en-ko-transliterator

作者 eunsour · 下载 34 · 点赞 2 · 更新 2023-10-31

transliteration multilingual
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

def transliteration(word: str):
    model_checkpoint = "eunsour/en-ko-transliterator"
    
    model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
    tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, src_lang="en", tgt_lang="ko")
    
    encoded_en = tokenizer(word, truncation=True, max_length=48, return_tensors="pt")
    generated_tokens = model.generate(**encoded_en)
    result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
    
    return result

transliteration("transformer")
# ['트랜스포머']

模型信息

该模型卡暂无预览图和文字说明。以下信息来自 Hugging Face。

作者 eunsour
类型—
基础模型—
标签transliteration、multilingual
下载34
点赞2
更新时间2023-10-31
Hugging Face https://huggingface.co/eunsour/en-ko-transliterator