yelyah commited on
Commit
b98ba60
1 Parent(s): 6ae9cf1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +36 -2
README.md CHANGED
@@ -13,6 +13,40 @@ tags:
13
 
14
  # Model Card for Model ID
15
 
16
- <!-- Provide a quick summary of what the model is/does. -->
17
- The mT5-multilingual-XLSum model was fine-tuned on the UA-News dataset to generate news headlines in Ukrainian language.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
 
13
 
14
  # Model Card for Model ID
15
 
16
+ ## Model Summary
17
+ The mT5-multilingual-XLSum model was fine-tuned on the UA-News dataset to generate news headlines in Ukrainian language. This model is designed to produce concise and accurate summaries for Ukrainian news articles.
18
+
19
+ ## Model Details
20
+ - **Model Type**: Summarization
21
+ - **Language**: Multilingual
22
+ - **Library**: Transformers
23
+
24
+ ## Dataset
25
+ - **Name**: UA-News
26
+ - **Source**: FIdo-AI
27
+ - **Description**: The UA-News dataset contains a diverse collection of Ukrainian news articles, covering various topics including politics, economics, culture, and sports. The dataset is curated to provide high-quality training data for summarization tasks.
28
+
29
+ ## Training
30
+ - **Fine-Tuning**: The model was fine-tuned on the UA-News dataset using the Hugging Face Transformers library.
31
+ - **Epochs**: 4
32
+ - **Batch Size**: 4
33
+ - **Learning Rate**: 4e-5
34
+
35
+ ## Evaluation
36
+ - **Metrics**: The model's performance was evaluated using the ROUGE metric.
37
+ - **ROUGE-1**: 0.2452
38
+ - **ROUGE-2**: 0.1075
39
+ - **ROUGE-L**: 0.2348
40
+ - **BERTScore**: 0.7573
41
+
42
+ ## Usage
43
+ - **Pipeline Tag**: Summarization
44
+ - **How to Use**: The model can be used with the Hugging Face `pipeline` for summarization. Here's an example:
45
+ ```python
46
+ from transformers import pipeline
47
+
48
+ summarizer = pipeline("summarization", model="yelyah/mT5-XLSUM-ua-news ")
49
+ article = "Your news article text here."
50
+ summary = summarizer(article)
51
+ print(summary)
52