andrewbrown commited on
Commit
f782642
·
1 Parent(s): 4b52e01

update readme with example prompt and proper liscencing

Browse files
Files changed (1) hide show
  1. README.md +23 -8
README.md CHANGED
@@ -1,25 +1,40 @@
1
- ## GPT2-MI-Reflector
 
 
 
 
 
 
 
 
2
 
3
  ### A gpt2 based motivational interviewing reflector
4
 
 
 
 
 
5
  To properly inference model (as it was fine-tuned), use the following format:
6
 
7
  -------------------------------------------------------------------------------------------------------------------
8
- Prompt: ...Question from bot...
9
 
10
- Response: ...Response from user...\n
11
 
12
  -------------------------------------------------------------------------------------------------------------------
13
 
 
 
 
14
 
 
15
 
16
- Model should generate a motivational interviewing adherent statement
17
 
18
 
19
 
20
- First model upload to huggingface!
 
21
 
22
- ---
23
- license: openrail
24
- ---
25
 
 
 
1
+ ---
2
+ license: openrail
3
+ language:
4
+ - en
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - medical
8
+ ---
9
+ # GPT2-MI-Reflector
10
 
11
  ### A gpt2 based motivational interviewing reflector
12
 
13
+ Given a prompt and a response, this GPT2 based transformer will generate a reflection. It was fine-tuned on prompt/response/reflection triplets so it should also be inferenced with the same input style. Examples are shown below.
14
+
15
+ The triplets this reflector was fine-tuned on focus on smoking-cessation. Given this, its best performance is when inferencing about smoking.
16
+
17
  To properly inference model (as it was fine-tuned), use the following format:
18
 
19
  -------------------------------------------------------------------------------------------------------------------
20
+ <|startoftext|> Prompt: ...Question from bot...
21
 
22
+ Response: ...Response from user...
23
 
24
  -------------------------------------------------------------------------------------------------------------------
25
 
26
+ Example Prompt: paste two lines below and generate a reflection
27
+ -------------------------------------------------------------------------------------------------------------------
28
+ <|startoftext|>Prompt: What do you want to change in your smoking?
29
 
30
+ Response: I want to smoke less often in my life, its become quite stressful for me.
31
 
32
+ -------------------------------------------------------------------------------------------------------------------
33
 
34
 
35
 
36
+ Model should generate a motivational interviewing adherent statement
37
+
38
 
 
 
 
39
 
40
+ This is the first model (of many!) that I'll upload to huggingface!