from transformers import pipeline, T5TokenizerFast

tokenizer_name = "paust/pko-t5-base"
tokenizer = T5TokenizerFast.from_pretrained(tokenizer_name)
model_path = 'Darong/BlueT'
translator = pipeline("translation", model=model_path, tokenizer=tokenizer, max_length=255)
# 영어 -> 한국어
prefix = "E2K: "
source = "This model is an English-Korean translation model."
target = translator(prefix + source)
print(target[0]['translation_text'])

# 한국어->영어
prefix = "K2E: "
source = "이 모델은 영어-한국어 번역 모델입니다."
target = translator(prefix + source)
print(target[0]['translation_text'])

Training Details

Training Data

이 모델은 AI Hub 및 자체 구축한 데이터로 학습되었습니다. 영어->한국어 학습 데이터 수는 1800만 이상, 한국어->영어 학습 데이터 수는 1200만 이상의 문장으로 구축되었습니다.