chieunq/vietnamese-sentence-paraphase-v1

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.
----------------------------------------------------------------------------------------------------