tngo commited on
Commit
9b0ab3a
1 Parent(s): 01bdfbf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -12,8 +12,11 @@ This is an instance of microsoft/DialoGPT-small trained on a tv show character,
12
  Occasionally the chatbot just responds with just multiple '!' characters. The chatbot frequently responds with "I'm not your buddy, pal" to uncomfortable/strange prompts/messages. Still working on a fix for those known issues.
13
 
14
  ```Python
15
- tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-small')
16
- model = AutoModelWithLMHead.from_pretrained('output-small')
 
 
 
17
 
18
  # Let's chat for 4 lines
19
  for step in range(4):
 
12
  Occasionally the chatbot just responds with just multiple '!' characters. The chatbot frequently responds with "I'm not your buddy, pal" to uncomfortable/strange prompts/messages. Still working on a fix for those known issues.
13
 
14
  ```Python
15
+ from transformers import AutoTokenizer, AutoModelWithLMHead
16
+
17
+ tokenizer = AutoTokenizer.from_pretrained("tngo/DialoGPT-small-HankHill")
18
+
19
+ model = AutoModelWithLMHead.from_pretrained("tngo/DialoGPT-small-HankHill")
20
 
21
  # Let's chat for 4 lines
22
  for step in range(4):