Downloads · 30 days
19
1% of all-time downloads
ukr-models/uk-summarizer
uk-summarizer is a machine learning model from ukr-models. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
Fine-tuning of uk-mt5-base model on summarization dataset.
Downloads · 30 days
19
1% of all-time downloads
All-time downloads
3.1K
Public
Parameters
244M
2 GB on disk
Likes
4
Public
Click a slice to open those files.
.bin977 MB · 50%
From the Hugging Face model README
Fine-tuning of uk-mt5-base model on summarization dataset.
from transformers import AutoTokenizer, T5ForConditionalGeneration, pipeline
tokenizer = AutoTokenizer.from_pretrained('ukr-models/uk-summarizer')
model = T5ForConditionalGeneration.from_pretrained('ukr-models/uk-summarizer')
ppln = pipeline("summarization", model=model, tokenizer=tokenizer, device=0, max_length=128, num_beams=4, no_repeat_ngram_size=2, clean_up_tokenization_spaces=True)
text = "..."
ppln(text)