yelyah/mT5-XLSUM-ua-news

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.

Dataset

• Name: UA-News
• Source: FIdo-AI

Training

• Fine-Tuning: The model was fine-tuned on the UA-News dataset using the Hugging Face Transformers library.
• 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)