Views
No views yet
1CKPT = 'chieunq/vietnamese-sentence-paraphase'
2from transformers import MT5Tokenizer, MT5ForConditionalGeneration
3tokenizer = MT5Tokenizer.from_pretrained(CKPT)
4model = MT5ForConditionalGeneration.from_pretrained(CKPT)
5def paraphase(text):
6 inputs = tokenizer(text, padding='longest', max_length=64, return_tensors='pt')
7 input_ids = inputs.input_ids
8 attention_mask = inputs.attention_mask
9 output = model.generate(input_ids, attention_mask=attention_mask, max_length=64)
10 return tokenizer.decode(output[0], skip_special_tokens=True)
11
12texts = [
13 "Thật tự hào khi là sinh viên trường Đại học Công nghệ",
14 "Cách kiếm nhiều tiền ?",
15 "Những nguyên lí cơ bản của vật lý ?",
16 "Làm thế nào để học ngôn ngữ Java",
17 "Ngoài ra, nắng nóng còn có thể gây tình trạng mất nước, kiệt sức, đột qụy do sốc nhiệt đối với cơ thể người khi tiếp xúc lâu với nền nhiệt độ cao."
18 ]
19for text in texts:
20 print(f'Input: {text}')
21 print(f'Output: {paraphase(text)}')
22 print('-'*100)
23Input: Thật tự hào khi là sinh viên trường Đại học Công nghệ
Output: Là sinh viên Đại học Công nghệ, tôi rất tự hào về điều đó.
----------------------------------------------------------------------------------------------------
Input: Cách kiếm nhiều tiền ?
Output: Một số cách để kiếm được nhiều tiền là gì?
----------------------------------------------------------------------------------------------------
Input: Những nguyên lí cơ bản của vật lý ?
Output: Các nguyên tắc cơ bản của vật lý là gì?
----------------------------------------------------------------------------------------------------
Input: Làm thế nào để học ngôn ngữ Java
Output: Các bước để thành thạo ngôn ngữ Java là gì?
----------------------------------------------------------------------------------------------------
Input: Ngoài ra, nắng nóng còn có thể gây tình trạng mất nước, kiệt sức, đột qụy do sốc nhiệt đối với cơ thể người khi tiếp xúc lâu với nền nhiệt độ cao.
Output: Hơn nữa, nắng nóng có thể dẫn đến mất nước, kiệt sức, đột quỵ do sốc nhiệt đối với cơ thể người khi tiếp xúc lâu với nền nhiệt độ cao.
----------------------------------------------------------------------------------------------------