metadata
language:
- vi
tags:
- vi-T5
- paraphase
- sentence-paraphase
This is tool for paraphase Vietnamese sentence.
How to run
CKPT = 'chieunq/vietnamese-sentence-paraphase-v1'
from transformers import MT5Tokenizer, MT5ForConditionalGeneration
tokenizer = MT5Tokenizer.from_pretrained(CKPT)
model = MT5ForConditionalGeneration.from_pretrained(CKPT)
def paraphase(text):
inputs = tokenizer(text, padding='longest', max_length=64, return_tensors='pt')
input_ids = inputs.input_ids
attention_mask = inputs.attention_mask
output = model.generate(input_ids, attention_mask=attention_mask, max_length=64)
return tokenizer.decode(output[0], skip_special_tokens=True)
texts = ["Làm sao để sống tốt đời đẹp đạo",
"Bài học tốt nhất trong cuộc sống là gì?",
"Các bước chuẩn bị phỏng vấn tại Google",
"Vì sao phụ nữ thường sống thọ hơn đàn ông?",
"Nắng nóng bắt đầu xảy ra ở nhiều nơi tại Bắc Bộ và sẽ tiếp tục tăng nhiệt trong vài ngày tới.",
"Trường Đại học Công nghệ luôn đào tạo ra những sinh viên tuyệt vời, có ích cho xã hội",
"Ô tô này có kích thước nhỏ và giá bán ở nước ngoài chưa đến trăm triệu.",
"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."
]
for text in texts:
print(f'Input: {text}')
print(f'Output: {paraphase(text)}')
print('-'*100)
Output
Input: Làm sao để sống tốt đời đẹp đạo
Output: Làm thế nào để tôi sống tốt đẹp
----------------------------------------------------------------------------------------------------
Input: Bài học tốt nhất trong cuộc sống là gì?
Output: Bài học cuộc sống tốt nhất là gì?
----------------------------------------------------------------------------------------------------
Input: Các bước chuẩn bị phỏng vấn tại Google
Output: Các bước chuẩn bị cho cuộc phỏng vấn tại Google là gì?
----------------------------------------------------------------------------------------------------
Input: Vì sao phụ nữ thường sống thọ hơn đàn ông?
Output: Tại sao phụ nữ sống thọ hơn đàn ông?
----------------------------------------------------------------------------------------------------
Input: Nắng nóng bắt đầu xảy ra ở nhiều nơi tại Bắc Bộ và sẽ tiếp tục tăng nhiệt trong vài ngày tới.
Output: Nắng nóng bắt đầu xảy ra ở Bắc Bộ và sẽ tiếp tục tăng nhiệt trong vài ngày tới.
----------------------------------------------------------------------------------------------------
Input: Trường Đại học Công nghệ luôn đào tạo ra những sinh viên tuyệt vời, có ích cho xã hội
Output: Trường Đại học Công nghệ luôn đào tạo những sinh viên tốt, có ích cho xã hội
----------------------------------------------------------------------------------------------------
Input: Ô tô này có kích thước nhỏ và giá bán ở nước ngoài chưa đến trăm triệu.
Output: ô tô này có kích thước nhỏ và giá bán ở nước ngoài không đến trăm triệu.
----------------------------------------------------------------------------------------------------
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: Ngoài ra, nắng nóng còn có thể gây ra tình trạng 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 với nhiệt độ cao.
----------------------------------------------------------------------------------------------------