soProf1998
commited on
Commit
•
ec21438
1
Parent(s):
b1290fe
Update README.md
Browse files
README.md
CHANGED
@@ -19,8 +19,8 @@ Chat with the model:
|
|
19 |
```python
|
20 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
21 |
|
22 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
23 |
-
model = AutoModelWithLMHead.from_pretrained("
|
24 |
# Let's chat for 4 lines
|
25 |
for step in range(4):
|
26 |
# encode the new user input, add the eos_token and return a tensor in Pytorch
|
@@ -40,5 +40,5 @@ for step in range(4):
|
|
40 |
)
|
41 |
|
42 |
# pretty print last ouput tokens from bot
|
43 |
-
print("
|
44 |
```
|
|
|
19 |
```python
|
20 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
21 |
|
22 |
+
tokenizer = AutoTokenizer.from_pretrained("soProf1998/DialoGPT-medium-chattyrick")
|
23 |
+
model = AutoModelWithLMHead.from_pretrained("soProf1998/DialoGPT-medium-chattyrick")
|
24 |
# Let's chat for 4 lines
|
25 |
for step in range(4):
|
26 |
# encode the new user input, add the eos_token and return a tensor in Pytorch
|
|
|
40 |
)
|
41 |
|
42 |
# pretty print last ouput tokens from bot
|
43 |
+
print("RickBot: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
|
44 |
```
|