|
--- |
|
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." |
|
] |
|
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. |
|
---------------------------------------------------------------------------------------------------- |
|
``` |