Update README.md

README.md

@@ -11,7 +11,7 @@ license:
 - apache-2.0
 inference:
   parameters:
-    repetition_penalty: 2.9
+    no_repeat_ngram_size: 4
 widget:
 - text: "Высота башни составляет 324 метра (1063 фута), примерно такая же высота, как у 81-этажного здания, и самое высокое сооружение в Париже. Его основание квадратно, размером 125 метров (410 футов) с любой стороны. Во время строительства Эйфелева башня превзошла монумент Вашингтона, став самым высоким искусственным сооружением в мире, и этот титул она удерживала в течение 41 года до завершения строительство здания Крайслер в Нью-Йорке в 1930 году. Это первое сооружение которое достигло высоты 300 метров. Из-за добавления вещательной антенны на вершине башни в 1957 году она сейчас выше здания Крайслер на 5,2 метра (17 футов). За исключением передатчиков, Эйфелева башня является второй самой высокой отдельно стоящей структурой во Франции после виадука Мийо."
   example_title: "Википедия"
@@ -54,7 +54,7 @@ input_ids = tokenizer(
 
 output_ids = model.generate(
     input_ids=input_ids,
-    repetition_penalty=3.0
+    no_repeat_ngram_size=4
 )[0]
 
 summary = tokenizer.decode(output_ids, skip_special_tokens=True)
@@ -72,20 +72,30 @@ print(summary)
 
 ## Eval results
 
-Gazeta v1 train -> Gazeta v1 test
-
-| Model                     | R-1-f | R-2-f | R-L-f | chrF | METEOR | BLEU |
-|:--------------------------|:------|:------|:------|:-------|:-------|:-----|
-| [mbart_ru_sum_gazeta](https://huggingface.co/IlyaGusev/mbart_ru_sum_gazeta)       | 32.1  | 14.3  | **27.9**  | **39.7** | **25.7**   | **12.4** |
-| [rut5_base_sum_gazeta](https://huggingface.co/IlyaGusev/rut5_base_sum_gazeta)      | **32.3**  | **14.5**  | 27.9 | 39.6 | 25.1   | 11.5 |
-
-Gazeta v1 train -> Gazeta v2 test
-
-| Model                     | R-1-f | R-2-f | R-L-f | chrF | METEOR | BLEU |
-|:--------------------------|:------|:------|:------|:-------|:-------|:-----|
-| [mbart_ru_sum_gazeta](https://huggingface.co/IlyaGusev/mbart_ru_sum_gazeta)        | **28.8**  | **11.1**  | **24.5**  | **37.4** | **22.7**   | **9.5** |
-| [rut5_base_sum_gazeta](https://huggingface.co/IlyaGusev/rut5_base_sum_gazeta)      | 28.5  | 11.0  | 24.2  | 36.8 | 21.2   | 8.6 |
-
+* Train dataset: **Gazeta v1 train**
+* Test dataset: **Gazeta v1 test**
+* Source max_length: **600**
+* Target max_length: **200**
+* no_repeat_ngram_size: **4**
+* num_beams: **5**
+
+| Model                     | R-1-f | R-2-f | R-L-f | chrF | METEOR | BLEU | Avg char length |
+|:--------------------------|:------|:------|:------|:-------|:-------|:-----|:-----|
+| [mbart_ru_sum_gazeta](https://huggingface.co/IlyaGusev/mbart_ru_sum_gazeta)       | **32.4**  | 14.3  | 28.0  | 39.7 | **26.4** | 12.1 | 371 |
+| [rut5_base_sum_gazeta](https://huggingface.co/IlyaGusev/rut5_base_sum_gazeta)      | 32.2  | **14.4**  | **28.1** | **39.8** | 25.7 | **12.3** | 330 |
+ |
+
+* Train dataset: **Gazeta v1 train**
+* Test dataset: **Gazeta v2 test**
+* Source max_length: **600**
+* Target max_length: **200**
+* no_repeat_ngram_size: **4**
+* num_beams: **5**
+
+| Model                     | R-1-f | R-2-f | R-L-f | chrF | METEOR | BLEU | Avg char length |
+|:--------------------------|:------|:------|:------|:-------|:-------|:-----|:-----|
+| [mbart_ru_sum_gazeta](https://huggingface.co/IlyaGusev/mbart_ru_sum_gazeta)        | **28.7**  | **11.1**  | 24.4  | **37.3** | **22.7**  | **9.4** | 373 |
+| [rut5_base_sum_gazeta](https://huggingface.co/IlyaGusev/rut5_base_sum_gazeta)      | 28.6 | **11.1** | **24.5** | 37.2 | 22.0 | **9.4** | 331 |
 
 Predicting all summaries:
 ```python