Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("PiotrWarzachowski/polish-text-summarizer")
4model = AutoModelForSeq2SeqLM.from_pretrained("PiotrWarzachowski/polish-text-summarizer")
5
6article = "Twój długi artykuł po polsku..."
7
8inputs = tokenizer(article, max_length=512, truncation=True, return_tensors="pt")
9outputs = model.generate(**inputs, max_length=128, num_beams=4, no_repeat_ngram_size=3)
10summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
11
12print(summary)