Update README.md
Browse files
README.md
CHANGED
@@ -14,32 +14,35 @@ pipeline_tag: text2text-generation
|
|
14 |
---
|
15 |
|
16 |
|
17 |
-
Model Overview
|
18 |
|
19 |
HealthModel_Qwen2.5-0.5B-Instruct is a fine-tuned version of Qwen/Qwen2.5-0.5B-Instruct, specifically trained to provide supportive and informative responses for mental health counseling conversations. This model is designed to assist users in seeking advice, coping strategies, and general guidance on mental well-being.
|
20 |
|
21 |
-
Fine-Tuning Details
|
22 |
|
23 |
This model has been further fine-tuned on the following datasets:
|
24 |
|
25 |
-
Amod/Mental Health Counseling Conversations: A dataset containing counseling conversations designed to support individuals struggling with mental health issues.
|
26 |
|
27 |
-
ZahrizhalAli/Mental Health Conversational Dataset: A dataset featuring mental health-related dialogues, aimed at enhancing conversational AI models for mental health support.
|
28 |
|
29 |
-
|
|
|
30 |
|
31 |
Loading the Model
|
32 |
|
33 |
To use the model, install transformers and torch if you haven’t already:
|
34 |
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
|
|
|
|
43 |
|
44 |
Generating Responses
|
45 |
|
|
|
14 |
---
|
15 |
|
16 |
|
17 |
+
# Model Overview
|
18 |
|
19 |
HealthModel_Qwen2.5-0.5B-Instruct is a fine-tuned version of Qwen/Qwen2.5-0.5B-Instruct, specifically trained to provide supportive and informative responses for mental health counseling conversations. This model is designed to assist users in seeking advice, coping strategies, and general guidance on mental well-being.
|
20 |
|
21 |
+
## Fine-Tuning Details
|
22 |
|
23 |
This model has been further fine-tuned on the following datasets:
|
24 |
|
25 |
+
Amod/Mental Health Counseling Conversations: A dataset containing counseling conversations designed to support individuals struggling with mental health issues.
|
26 |
|
27 |
+
ZahrizhalAli/Mental Health Conversational Dataset: A dataset featuring mental health-related dialogues, aimed at enhancing conversational AI models for mental health support.
|
28 |
|
29 |
+
|
30 |
+
## Model Usage
|
31 |
|
32 |
Loading the Model
|
33 |
|
34 |
To use the model, install transformers and torch if you haven’t already:
|
35 |
|
36 |
+
```sh
|
37 |
+
pip install transformers torch
|
38 |
+
|
39 |
+
Then, load the model and tokenizer:
|
40 |
+
|
41 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
42 |
+
|
43 |
+
tokenizer = AutoTokenizer.from_pretrained("your-hf-username/HealthModel_Qwen2.5-0.5B-Instruct")
|
44 |
+
model = AutoModelForCausalLM.from_pretrained("your-hf-username/HealthModel_Qwen2.5-0.5B-Instruct")
|
45 |
+
```
|
46 |
|
47 |
Generating Responses
|
48 |
|