Views
No views yet
1from transformers import (
2 AutoTokenizer,
3 AutoModelForSeq2SeqLM)
4
5tokenizer = AutoTokenizer.from_pretrained("bluexmas/mbart50_ko_vi")
6model = AutoModelForSeq2SeqLM.from_pretrained("bluexmas/mbart50_ko_vi")
7
8def translate(text):
9 batch = tokenizer.prepare_seq2seq_batch([text], return_tensors="pt", max_length=128, truncation=True)
10 translated = model.generate(**batch)
11 return tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
12
13text = "안녕하세요. 오늘 날씨는 어때요?"
14print(text, "--->", translate(text))
15# in chào. Hôm nay thời tiết thế nào ạ?
16
17text = "오늘 날씨가 정말 좋네요."
18print(text, "--->", translate(text))
19# Hôm nay thời tiết đẹp thật đấy.
20
21text = "모든 국민은 법 앞에 평등하다. 누구든지 성별ㆍ종교 또는 사회적 신분에 의하여 정치적ㆍ경제적ㆍ사회적ㆍ문화적 생활의 모든 영역에 있어서 차별을 받지 아니한다."
22print(text, "--->", translate(text))
23# Tất cả người dân đều bình đẳng trước pháp luật. Bất kỳ ai cũng không bị phân biệt đối xử trong mọi lĩnh vực đời sống chính trị, kinh tế, xã hội, văn hóa dựa trên giới tính, tôn giáo hay thân phận xã hội.