Update README.md
Browse files
README.md
CHANGED
@@ -10,8 +10,8 @@ tags: []
|
|
10 |
# Load model directly
|
11 |
from transformers import AutoModelForCausalLM, GPT2Tokenizer
|
12 |
|
13 |
-
tokenizer =
|
14 |
-
model =
|
15 |
```
|
16 |
|
17 |
### Chatting
|
|
|
10 |
# Load model directly
|
11 |
from transformers import AutoModelForCausalLM, GPT2Tokenizer
|
12 |
|
13 |
+
tokenizer = GPT2Tokenizer.from_pretrained("suriya7/conversational-gpt-1")
|
14 |
+
model = AutoModelForCausalLM.from_pretrained("suriya7/conversational-gpt-1")
|
15 |
```
|
16 |
|
17 |
### Chatting
|