kingabzpro
/

DialoGPT-small-Rick-Bot

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

kingabzpro commited on Aug 27, 2021

Commit

98c9a6b

·

1 Parent(s): f754e38

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ metrics:
 ```python
 tokenizer = AutoTokenizer.from_pretrained('kingabzpro/DialoGPT-small-Rick-Bot')
-model = AutoModelWithLMHead.from_pretrained(kingabzpro/DialoGPT-small-Rick-Bot')
 # Let's chat for 4 lines
 for step in range(4):
     # encode the new user input, add the eos_token and return a tensor in Pytorch

 ```python
 tokenizer = AutoTokenizer.from_pretrained('kingabzpro/DialoGPT-small-Rick-Bot')
+model = AutoModelWithLMHead.from_pretrained('kingabzpro/DialoGPT-small-Rick-Bot')
 # Let's chat for 4 lines
 for step in range(4):
     # encode the new user input, add the eos_token and return a tensor in Pytorch