Views
No views yet
1from transformers import AutoTokenizer, T5ForConditionalGeneration
2
3model_name = 'cabir40/t5-v1.1-base-dutch-cased_inversion'
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6
7model = T5ForConditionalGeneration.from_pretrained(model_name)1document =["Zonder relatie mensen zijn gelukkig?",
2 "Nu steeds meer Nederlanders worden ouder dan 100 jaar.",
3 "Gewoon ik open mijn ogen wijd, zodat het lijkt of ik goed luister.",
4 "Dan het wordt moeilijk, als anderen beginnen over andere dingen te praten",
5]
6
7inputs = tokenizer(document, return_tensors="pt", padding=True)
8
9output_sequences = model_from_local.generate(
10 input_ids=inputs["input_ids"],
11 attention_mask=inputs["attention_mask"] )
12
13tokenizer.batch_decode(output_sequences, skip_special_tokens=True)1['Zonder relatie zijn mensen gelukkig?',
2 'Nu worden steeds meer Nederlanders ouder dan 100 jaar.',
3 'Gewoon open ik mijn ogen wijd, zodat het lijkt of ik goed luister.',
4 'Dan wordt het moeilijk, als anderen beginnen over andere dingen te praten Dan']