thors/RRN
Viewer • Updated • 3.96k • 13 • 1
How to use thors/mt5-base-icelandic-summarization with Transformers:
# Use a pipeline as a high-level helper
# Warning: Pipeline type "summarization" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline
pipe = pipeline("summarization", model="thors/mt5-base-icelandic-summarization") # Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("thors/mt5-base-icelandic-summarization")
model = AutoModelForSeq2SeqLM.from_pretrained("thors/mt5-base-icelandic-summarization")# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("thors/mt5-base-icelandic-summarization")
model = AutoModelForSeq2SeqLM.from_pretrained("thors/mt5-base-icelandic-summarization")This is Google's mT5 base model fine-tuned for abstractive text summarization in Icelandic. The model is fine-tuned on the RÚV Radio News (RRN) dataset.
@inproceedings{sverrisson2023abstractive,
title={Abstractive Text Summarization for Icelandic},
author={Sverrisson, {\TH}{\'o}r and Einarsson, Hafsteinn},
booktitle={Proceedings of the 24th Nordic Conference on Computational Linguistics (NoDaLiDa)},
pages={17--31},
year={2023}
}
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("summarization", model="thors/mt5-base-icelandic-summarization")