Update README.md

README.md

@@ -1,12 +1,20 @@
 ---
 library_name: transformers
-tags: []
+datasets:
+- ShivomH/Mental-Health-Conversations
+- kamruzzaman-asif/reddit-mental-health-classification
+- jsfactory/mental_health_reddit_posts
+- ZahrizhalAli/mental_health_conversational_dataset
+language:
+- en
+base_model:
+- meta-llama/Meta-Llama-3-8B-Instruct
 ---
 
 
 # Mental Health Therapy Chatbot
 
-**Model Name**: `tanusrich/Mental_Health_Chatbot`  
+**Model Name**: `menzo-ai/mental-health-chatbot`  
 **Model Type**: LLaMA-based model fine-tuned for Mental Health Therapy assistance
 
 ## Overview
@@ -17,14 +25,14 @@ It is ideal for use in mental health support applications where users can receiv
 
 ## Model Details
 
-- **Base Model**: [LLaMA-2-7b](https://huggingface.co/meta-llama/Llama-2-7b)
+- **Base Model**: [Llama-3-8B](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct)
 - **Fine-Tuned Dataset**: A specialized dataset curated for mental health conversations.
 - **Purpose**: Provide empathetic, supportive responses for individuals seeking mental health assistance.
 - **Language**: English
 
 ## Model Architecture
 
-This model is based on **LLaMA-2-7b** architecture. It is a causal language model (CausalLM), which generates responses based on the input prompt by predicting the next word in the sequence.
+This model is based on **Llama-3-8B** architecture. It is a causal language model (CausalLM), which generates responses based on the input prompt by predicting the next word in the sequence.
 
 ### Model Capabilities
 
@@ -44,7 +52,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Load the model and tokenizer
-model_name = "tanusrich/Mental_Health_Chatbot"
+model_name = "menzo-ai/mental-health-chatbot"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
@@ -119,17 +127,16 @@ Mental health is a sensitive area, and while this model attempts to provide thou
 If you use this model, please cite the LLaMA-2 model and the fine-tuning process as follows:
 
 ```plaintext
-@article{LLaMA2,
-  title={LLaMA 2: Open Foundation and Fine-Tuned Chat Models},
+@article{LLaMA-3-8B,
+  title={LLaMA-3-8B: Open Foundation and Fine-Tuned Chat Models},
   author={Meta AI},
-  year={2023}
+  year={2025}
 }
 
-@misc{tanusrich2024MentalHealthChatbot,
-  author = {Tanusri},
+@misc{menzoai2025MentalHealthChatbot,
+  author = {menzo-ai},
   title = {Mental Health Therapy Chatbot},
   year = {2024},
-  url = {https://huggingface.co/tanusrich/Mental_Health_Chatbot}
+  url = {https://huggingface.co/menzo-ai/mental-health-chatbot/}
 }
-```
-
+```