Update README.md
Browse files
README.md
CHANGED
|
@@ -27,8 +27,8 @@ This model is a GPT-2 model fine-tuned on the SQuAD (Stanford Question Answering
|
|
| 27 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
| 28 |
|
| 29 |
# Load model and tokenizer
|
| 30 |
-
model = GPT2LMHeadModel.from_pretrained("
|
| 31 |
-
tokenizer = GPT2Tokenizer.from_pretrained("
|
| 32 |
|
| 33 |
# Prepare input
|
| 34 |
context = "George Washington was the first president of the United States, serving from 1789 to 1797."
|
|
|
|
| 27 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
| 28 |
|
| 29 |
# Load model and tokenizer
|
| 30 |
+
model = GPT2LMHeadModel.from_pretrained("houcine-bdk/chatMachine_v1")
|
| 31 |
+
tokenizer = GPT2Tokenizer.from_pretrained("houcine-bdk/chatMachine_v1")
|
| 32 |
|
| 33 |
# Prepare input
|
| 34 |
context = "George Washington was the first president of the United States, serving from 1789 to 1797."
|