Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2tok = AutoTokenizer.from_pretrained("HoangQuocViet25/nlp-btl-wikilingua-x5plus2")
3net = AutoModelForSeq2SeqLM.from_pretrained("HoangQuocViet25/nlp-btl-wikilingua-x5plus2").eval()
4# ViT5's own preprocessing: no task prefix, EOS appended to the raw string.
5enc = tok(article.strip() + "</s>", return_tensors="pt", truncation=True, max_length=1024)
6out = net.generate(**enc, num_beams=4, length_penalty=1.0, max_new_tokens=256,
7 early_stopping=True)
8print(tok.decode(out[0], skip_special_tokens=True))