Views
No views yet
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3model = T5ForConditionalGeneration.from_pretrained("huyg1108/t5-ancient_vie-eng-base")
4tokenizer = T5Tokenizer.from_pretrained("huyg1108/t5-ancient_vie-eng-base", legacy=False)
5model.eval()
6
7src = "Đạo chi vi vật, duy hoảng duy hốt."
8tokenized_text = tokenizer.encode(src, return_tensors="pt")
9summary_ids = model.generate(
10 tokenized_text,
11 max_length=64,
12 num_beams=7,
13 repetition_penalty=2.5,
14 length_penalty=1.0,
15 early_stopping=True
16)
17
18output = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
19output