Update README.md
Browse files
README.md
CHANGED
@@ -37,6 +37,7 @@ Pre-trained language models have attracted increasing attention in the biomedica
|
|
37 |
## Dataset
|
38 |
|
39 |
ChatDoctor-200K dataset is collected from this paper https://arxiv.org/pdf/2303.14070.pdf
|
|
|
40 |
The dataset is composed by:
|
41 |
|
42 |
- 100k real conversations between patients and doctors from HealthCareMagic.com [HealthCareMagic-100k](https://drive.google.com/file/d/1lyfqIwlLSClhgrCutWuEe_IACNq6XNUt/view?usp=sharing).
|
@@ -90,6 +91,20 @@ def generate(
|
|
90 |
s = generation_output.sequences[0]
|
91 |
output = tokenizer.decode(s, skip_special_tokens=True)
|
92 |
return output.split(" Response:")[1]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
93 |
```
|
94 |
|
95 |
## Citation
|
|
|
37 |
## Dataset
|
38 |
|
39 |
ChatDoctor-200K dataset is collected from this paper https://arxiv.org/pdf/2303.14070.pdf
|
40 |
+
|
41 |
The dataset is composed by:
|
42 |
|
43 |
- 100k real conversations between patients and doctors from HealthCareMagic.com [HealthCareMagic-100k](https://drive.google.com/file/d/1lyfqIwlLSClhgrCutWuEe_IACNq6XNUt/view?usp=sharing).
|
|
|
91 |
s = generation_output.sequences[0]
|
92 |
output = tokenizer.decode(s, skip_special_tokens=True)
|
93 |
return output.split(" Response:")[1]
|
94 |
+
|
95 |
+
example_prompt = """
|
96 |
+
Below is an instruction that describes a task, paired with an input that provides further context.Write a response that appropriately completes the request.
|
97 |
+
|
98 |
+
### Instruction:
|
99 |
+
If you are a doctor, please answer the medical questions based on the patient's description.
|
100 |
+
|
101 |
+
### Input:
|
102 |
+
Hi i have sore lumps under the skin on my legs. they started on my left ankle and are approx 1 - 2cm diameter and are spreading up onto my thies. I am eating panadol night and anti allergy pills (Atarax). I have had this for about two weeks now. Please advise.
|
103 |
+
|
104 |
+
### Response:
|
105 |
+
"""
|
106 |
+
|
107 |
+
print(generate(example_prompt))
|
108 |
```
|
109 |
|
110 |
## Citation
|