ritakurban commited on
Commit
dcb0998
1 Parent(s): 293b74d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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