Update README.md

README.md

@@ -1,18 +1,20 @@
 ---
 license: mit
 tags:
-  - text-generation
-  - dialogue-modeling
-  - mental-health
-  - therapy
-  - empathetic-ai
+- text-generation
+- dialogue-modeling
+- mental-health
+- therapy
+- empathetic-ai
 datasets:
-  - samdak93/EmpathicConversations
+- samdak93/EmpathicConversations
 language:
-  - en
+- en
 model-index:
-  - name: Solacia
-    results: []
+- name: Solacia
+  results: []
+base_model:
+- openai-community/gpt2
 ---
 
 # Solacia: An Empathic AI Therapy Assistant
@@ -25,7 +27,7 @@ model-index:
 - **Author**: [samdak93](https://huggingface.co/samdak93)
 - **Language**: English
 - **License**: MIT
-- **Base Model**: *[insert base model name, e.g., TinyGPT-2, Mistral-7B, etc.]*
+- **Base Model**: *[gpt2]*
 - **Training Dataset**: [EmpathicConversations](https://huggingface.co/datasets/samdak93/EmpathicConversations)
 
 ## Use Cases
@@ -62,4 +64,4 @@ If you use this model, please consider citing the dataset and repository:
   title = {Solacia: An Empathic AI Therapy Assistant},
   year = {2025},
   url = {https://huggingface.co/samdak93/solacia}
-}
+}