Views
No views yet
1from transformers import AutoTokenizer, T5ForConditionalGeneration, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained('ukr-models/uk-summarizer')
4model = T5ForConditionalGeneration.from_pretrained('ukr-models/uk-summarizer')
5
6ppln = pipeline("summarization", model=model, tokenizer=tokenizer, device=0, max_length=128, num_beams=4, no_repeat_ngram_size=2, clean_up_tokenization_spaces=True)
7
8text = "..."
9ppln(text)