Update README.md (#6)

- Update README.md (c5bf7d614faffd8447db66b7cdbdb315bb4e2cca)


Co-authored-by: Anton Shapkin <[email protected]>

README.md

@@ -1,5 +1,19 @@
 ---
 license: apache-2.0
+datasets:
+- JetBrains/KExercises
+results:
+  - task:
+      type: text-generation
+    dataset:
+      name: MultiPL-HumanEval (Kotlin)
+      type: openai_humaneval
+    metrics:
+      - name: pass@1
+        type: pass@1
+        value: 42.24
+tags:
+- code
 ---
 
 # Kexer models
@@ -37,6 +51,11 @@ generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
 print(generated_text)
 ```
 
+As with the base model, we can use FIM. To do this, the following format must be used: 
+```
+'<PRE> ' + prefix + ' <SUF> ' + suffix + ' <MID>'
+```
+
 # Training setup
 
 The model was trained on one A100 GPU with following hyperparameters:
@@ -62,10 +81,10 @@ To evaluate we used Kotlin Humaneval (more infromation here)
 
 Fine-tuned model:
 
-|         **Model name**           |             **Kotlin HumanEval Pass Rate**              |             **Kotlin Completion**              |
-|:---------------------------:|:----------------------------------------:|:----------------------------------------:|
-|           `base model`            |           26.89            |           0.388            |
-|        `fine-tuned model`        |          42.24         |          0.344          |
+|         **Model name**           |             **Kotlin HumanEval Pass Rate**              |
+|:---------------------------:|:----------------------------------------:|
+|           `base model`            |           26.89            |
+|        `fine-tuned model`        |          42.24         |
 
 # Ethical Considerations and Limitations