Update README.md

README.md

@@ -75,18 +75,19 @@ print(id2label[result[0]])
 
 The source material was the recordings of psychotherapeutic sessions posted on YouTube in the public domain. After conducting speaker diarization and transcription of the recordings 15324 items (sentences) were obtained.
 
-## Links
-
-- **Paper [optional]:** <https://arxiv.org/abs/2412.17449>
-- 
-
-
 ## Recommendations
 
+Given the broad context of issues discussed in psychotherapeutic sessions, the authors believe that this model can be used to analyze human communication in general.
 
 ## Metrics
 
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+Score metrics of trained model
+
+|Sample|F1 macro|
+|:-|:-:|
+|Test|0.74|
+|Validation|0.77|
+|Train|0.97|
 
 [More Information Needed]
 
@@ -94,15 +95,14 @@ The source material was the recordings of psychotherapeutic sessions posted on Y
 
 ## Citation 
 
-**Papers:** Applying LLM and Topic Modelling in Psychotherapeutic Contexts <https://arxiv.org/abs/2412.17449>
-
-**Developed by:** @myentity
-**License:** MIT
-**Finetuned from model:** google-bert/bert-base-uncased
+- **Papers:** Vanin, A., Bolshev, V., & Panfilova, A. (2024). Applying LLM and Topic Modelling in Psychotherapeutic Contexts. ArXiv, abs/2412.17449. <https://arxiv.org/abs/2412.17449>
+- **Developed by:** @myentity
+- **License:** MIT
 
 
 **BibTeX:**
 
+```
 @misc{vanin2024applyingllmtopicmodelling,
       title={Applying LLM and Topic Modelling in Psychotherapeutic Contexts}, 
       author={Alexander Vanin and Vadim Bolshev and Anastasia Panfilova},
@@ -112,6 +112,7 @@ The source material was the recordings of psychotherapeutic sessions posted on Y
       primaryClass={cs.LG},
       url={https://arxiv.org/abs/2412.17449}, 
 }
+```
 
 ## Model Card Authors [optional]