Update README.md

README.md

@@ -6,14 +6,17 @@ pipeline_tag: text-generation
 ---
 
 # DialoGPT-medium-spanish-chitchat
+
 ## Description
 
-This is a **transformer-decoder** [gpt2 model](https://huggingface.co/gpt2), adapted for **single turn dialogue tasks**. We fine-tuned a [DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium) model from Microsoft, following the CLM (Causal Language Modelling) objective.
+This is a **transformer-decoder** [gpt2 model](https://huggingface.co/gpt2), adapted for **single turn dialogue tasks**. We fine-tuned a 345M parameters [DialoGPT-medium](https://huggingface.co/microsoft/DialoGPT-medium) model from Microsoft, following the CLM (Causal Language Modelling) objective.
 We used one of the datasets available in the [Bot Framework Tools repository](https://github.com/microsoft/botframework-cli). We processed [the professional-styled personality chat dataset in Spanish](https://github.com/microsoft/botframework-cli/blob/main/packages/qnamaker/docs/chit-chat-dataset.md), the file is available [here](https://qnamakerstore.blob.core.windows.net/qnamakerdata/editorial/spanish/qna_chitchat_professional.tsv)
+
 ---
+
 ## Example inference script
 
-Check at this example script to run this model in inference mode:
+### Check at this example script to run this model in inference mode
 
 ```python
 import torch
@@ -39,9 +42,12 @@ for i in range(CHAT_TURNS):
   print(f"Step - {i} >> DialoGPT-spanish model answer -> {step_model_answer}")
 
 ```
+
 ---
+
 ## Examples
-* Single-turn generation examples from an interactive environment:
+
+### Single-turn generation examples from an interactive environment
 
 |Role | Response |
 |---------|--------|
@@ -59,6 +65,16 @@ for i in range(CHAT_TURNS):
 | Bot |  En realidad no tengo tanta gracia. |
 |User | Jajajajajajajaja |
 | Bot | ¡Me alegro de que te rías! |
+
+---
+
 ## Fine-tuning in different dataset
 
-For fine-tuning this model, you can start from the DialoGPT base model
+If you want to fine-tune this model, we recommend you to start from the [DialoGPT model](https://huggingface.co/microsoft/DialoGPT-medium).
+You can check the [original GitHub repository](https://github.com/microsoft/DialoGPT).
+
+## Limitations
+
+- This model is intended to be used **just for single-turn chitchat conversations in Spanish**
+- This model's generation capabilities are limited to the extent of the aforementioned fine-tuning dataset
+- This model generates short answers, providing general context dialogue in a professional style