skiddieofficial commited on
Commit
15840dd
·
verified ·
1 Parent(s): 336bf21

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -13
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
- Model Usage
 
30
 
31
  Loading the Model
32
 
33
  To use the model, install transformers and torch if you haven’t already:
34
 
35
- pip install transformers torch
36
-
37
- Then, load the model and tokenizer:
38
-
39
- from transformers import AutoModelForCausalLM, AutoTokenizer
40
-
41
- tokenizer = AutoTokenizer.from_pretrained("your-hf-username/HealthModel_Qwen2.5-0.5B-Instruct")
42
- model = AutoModelForCausalLM.from_pretrained("your-hf-username/HealthModel_Qwen2.5-0.5B-Instruct")
 
 
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