Commit
•
dcb0998
1
Parent(s):
293b74d
Update README.md
Browse files
README.md
CHANGED
@@ -13,8 +13,8 @@ You can use this model for medical question-answering tasks by simply loading it
|
|
13 |
```
|
14 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
15 |
|
16 |
-
tokenizer = GPT2Tokenizer.from_pretrained("DistilGPT_PubMedQA")
|
17 |
-
model = GPT2LMHeadModel.from_pretrained("DistilGPT_PubMedQA")
|
18 |
|
19 |
prompt = "question: What is the primary function of the liver? context: The liver is a vital organ that performs several essential functions, including detoxification, protein synthesis, and the production of biochemicals necessary for digestion."
|
20 |
|
|
|
13 |
```
|
14 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
15 |
|
16 |
+
tokenizer = GPT2Tokenizer.from_pretrained("ritakurban/DistilGPT_PubMedQA")
|
17 |
+
model = GPT2LMHeadModel.from_pretrained("ritakurban/DistilGPT_PubMedQA")
|
18 |
|
19 |
prompt = "question: What is the primary function of the liver? context: The liver is a vital organ that performs several essential functions, including detoxification, protein synthesis, and the production of biochemicals necessary for digestion."
|
20 |
|