Downloads · 30 days
27
1% of all-time downloads
ml6team/mbart-large-cc25-cnn-dailymail-nl
mbart-large-cc25-cnn-dailymail-nl is a summarization model from ml6team. Use it when you need a shorter version of a longer text. It is set up for transformers.
Finetuned version of mbart. We also wrote a blog post about this model here
Downloads · 30 days
27
1% of all-time downloads
All-time downloads
4.7K
Public
Repo size
4.9 GB
Likes
6
Public
Click a slice to open those files.
.bin2.4 GB · 100%
From the Hugging Face model README
Finetuned version of mbart. We also wrote a blog post about this model here
It's meant for summarizing Dutch news articles.
import transformers
undisputed_best_model = transformers.MBartForConditionalGeneration.from_pretrained(
"ml6team/mbart-large-cc25-cnn-dailymail-nl"
)
tokenizer = transformers.MBartTokenizer.from_pretrained("facebook/mbart-large-cc25")
summarization_pipeline = transformers.pipeline(
task="summarization",
model=undisputed_best_model,
tokenizer=tokenizer,
)
summarization_pipeline.model.config.decoder_start_token_id = tokenizer.lang_code_to_id[
"nl_XX"
]
article = "Kan je dit even samenvatten alsjeblief." # Dutch
summarization_pipeline(
article,
do_sample=True,
top_p=0.75,
top_k=50,
# num_beams=4,
min_length=50,
early_stopping=True,
truncation=True,
)[0]["summary_text"]
Finetuned mbart with this dataset