Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model_id = "nqp426/vit5-ai-rewrite"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
7
8text = "[REWRITE] Thuật toán học máy ngày càng phát triển mạnh mẽ."
9inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)
10outputs = model.generate(**inputs, max_new_tokens=256, num_beams=4)
11
12rewritten = tokenizer.decode(outputs[0], skip_special_tokens=True)
13print(rewritten)