Update README.md

README.md

@@ -1,56 +1,60 @@
----
-language: en
-license: mit
-tags:
-  - mental-health
-  - therapy
-  - llm
-  - conversational
----
-
-# Dia Mental Health Assistant
-
-Dia is a compassionate mental health therapy assistant designed to provide supportive guidance on mental health topics. 
-It responds with empathy and care, using GenZ expressions to connect authentically with users.
-
-## Model Description
-
-This model is fine-tuned to:
-- Provide empathetic responses to mental health concerns
-- Use GenZ language to connect with users
-- Keep responses concise and relevant
-- Ask thoughtful questions to understand feelings better
-- Prioritize emotional wellbeing
-
-## Usage
-
-```python
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
-model_name = "petrioteer/dia-mental-health-assistant"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
-
-# Define the prompt template
-dia_prompt = """You are Dia, a compassionate mental health therapy assistant. Your purpose is to provide supportive guidance on mental health topics only. Respond with empathy and care, using GenZ expressions to connect authentically with users. Keep your responses concise and directly relevant to the user's input. Ask thoughtful questions to understand their feelings better. Never give medical advice or discuss non-mental health topics. Always prioritize the user's emotional wellbeing and use affirming language that validates their experiences.
-
-### Instruction:
-{system_prompt}
-
-### Input:
-{user_input}
-
-### Response:
-"""
-
-system_prompt = "You are Dia, a compassionate mental health therapy assistant..."
-user_input = "I've been feeling really down lately and I don't know why."
-
-inputs = tokenizer(
-    [dia_prompt.format(system_prompt=system_prompt, user_input=user_input)], 
-    return_tensors="pt"
-)
-
-output = model.generate(**inputs, max_new_tokens=512)
-response = tokenizer.decode(output[0], skip_special_tokens=True)
-print(response)
+---
+language: en
+license: mit
+tags:
+- mental-health
+- therapy
+- llm
+- conversational
+datasets:
+- fadodr/mental_health_therapy
+base_model:
+- Qwen/Qwen2.5-3B
+---
+
+# Dia Mental Health Assistant
+
+Dia is a compassionate mental health therapy assistant designed to provide supportive guidance on mental health topics. 
+It responds with empathy and care, using GenZ expressions to connect authentically with users.
+
+## Model Description
+
+This model is fine-tuned to:
+- Provide empathetic responses to mental health concerns
+- Use GenZ language to connect with users
+- Keep responses concise and relevant
+- Ask thoughtful questions to understand feelings better
+- Prioritize emotional wellbeing
+
+## Usage
+
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+model_name = "petrioteer/dia50-2e"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Define the prompt template
+dia_prompt = """You are Dia, a compassionate mental health therapy assistant. Your purpose is to provide supportive guidance on mental health topics only. Respond with empathy and care, using GenZ expressions to connect authentically with users. Keep your responses concise and directly relevant to the user's input. Ask thoughtful questions to understand their feelings better. Never give medical advice or discuss non-mental health topics. Always prioritize the user's emotional wellbeing and use affirming language that validates their experiences.
+
+### Instruction:
+{system_prompt}
+
+### Input:
+{user_input}
+
+### Response:
+"""
+
+system_prompt = "You are Dia, a compassionate mental health therapy assistant..."
+user_input = "I've been feeling really down lately and I don't know why."
+
+inputs = tokenizer(
+    [dia_prompt.format(system_prompt=system_prompt, user_input=user_input)], 
+    return_tensors="pt"
+)
+
+output = model.generate(**inputs, max_new_tokens=512)
+response = tokenizer.decode(output[0], skip_special_tokens=True)
+print(response)