soProf1998 commited on
Commit
ec21438
1 Parent(s): b1290fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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("r3dhummingbird/DialoGPT-medium-joshua")
23
- model = AutoModelWithLMHead.from_pretrained("r3dhummingbird/DialoGPT-medium-joshua")
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("JoshuaBot: {}".format(tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)))
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
  ```