Edit model card

Model Card for Model ID

Model Summary

The mT5-multilingual-XLSum model was fine-tuned on the UA-News dataset to generate concise and accurate news headlines in Ukrainian language.

Training

  • Epochs: 4
  • Batch Size: 4
  • Learning Rate: 4e-5

Evaluation

  • Metrics: The model's performance on the test set.
    • ROUGE-1: 0.2452
    • ROUGE-2: 0.1075
    • ROUGE-L: 0.2348
    • BERTScore: 0.7573

Usage

  • Pipeline Tag: Summarization
  • How to Use: The model can be used with the Hugging Face pipeline for summarization. Here's an example:
    from transformers import pipeline
    
    summarizer = pipeline("summarization", model="yelyah/mT5-XLSUM-ua-news")
    article = "Your news article text here."
    summary = summarizer(article)
    print(summary)
    
Downloads last month
4
Safetensors
Model size
582M params
Tensor type
F32
·
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train yelyah/mT5-XLSUM-ua-news