Views
No views yet
1from transformers import pipeline, T5TokenizerFast
2
3tokenizer_name = "paust/pko-t5-base"
4tokenizer = T5TokenizerFast.from_pretrained(tokenizer_name)
5model_path = 'Darong/BlueGame'
6translator = pipeline("translation", model=model_path, tokenizer=tokenizer, max_length=255)
7# 영어 -> 한국어
8prefix = "E2K: "
9source = "This model is an English-Korean translation model."
10target = translator(prefix + source)
11print(target[0]['translation_text'])
12