Update README.md

README.md

@@ -1,6 +1,6 @@
+
 ---
-language:
-- ru
+language: ru
 tags:
 - text-classification
 - multi-label-classification
@@ -15,13 +15,13 @@ metrics:
 
 # Russian Emotion Classifier
 
-Multi-label классификатор эмоций для русскоязычных текстов.  
-Обучен на датасете [CEDR](https://huggingface.co/datasets/sagteam/cedr_v1) с взвешенным BCE loss для корректной работы с дисбалансом классов.
+A multi-label emotion classifier for Russian texts.  
+Trained on the [CEDR](https://huggingface.co/datasets/sagteam/cedr_v1) dataset with weighted BCE loss to handle class imbalance.
 
-**Базовая модель**: `cointegrated/rubert-tiny2`  
+**Base model**: `cointegrated/rubert-tiny2`  
 **F1-micro**: `0.7247` | **F1-macro**: `0.6823`
 
-## Использование
+## Usage
 
 ```python
 from transformers import pipeline
@@ -38,26 +38,45 @@ for r in sorted(results, key=lambda x: -x["score"]):
         print(r["label"], round(r["score"], 3))
 ```
 
-## Метрики
+Example output:
+```
+joy 0.892
+fear 0.678
+```
+
+## Metrics
 
-| Метрика | Значение |
-|---------|---------|
+| Metric | Value |
+|--------|-------|
 | F1 micro | 0.7247 |
 | F1 macro | 0.6823 |
 | F1 weighted | 0.7389 |
 | Precision micro | 0.6193 |
 | Recall micro | 0.8733 |
 
-### F1 по классам
+### Per-class F1
 
-| Класс | F1 |
-|-------|----|
+| Class | F1 |
+|-------|-----|
 | joy | 0.8391 |
 | sadness | 0.8057 |
 | surprise | 0.6711 |
 | fear | 0.6488 |
 | anger | 0.4468 |
 
-## Классы
+## Labels
 
 `joy` · `sadness` · `surprise` · `fear` · `anger`
+
+## Citation
+
+If you use this model in your research, please cite:
+
+```bibtex
+@dataset{cedr_v1,
+  author = {SAGTeam},
+  title = {CEDR: Russian Emotion Dataset},
+  year = {2023},
+  url = {https://huggingface.co/datasets/sagteam/cedr_v1}
+}
+```