nqp426/vit5-ai-rewrite

ViT5 AI Rewrite Model

(Mô hình viết lại văn bản tiếng Việt)

Giới thiệu (Overview)

Mô hình này được fine-tune từ VietAI/vit5-base để viết lại các đoạn văn bản nhằm né tránh hệ thống phát hiện AI (AI Detectors), nhưng vẫn giữ nguyên ý nghĩa gốc.

Sử dụng (Usage)

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

model_id = "nqp426/vit5-ai-rewrite"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSeq2SeqLM.from_pretrained(model_id)

text = "[REWRITE] Thuật toán học máy ngày càng phát triển mạnh mẽ."
inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)
outputs = model.generate(**inputs, max_new_tokens=256, num_beams=4)

rewritten = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(rewritten)