chujiezheng
commited on
Commit
•
84dc62a
1
Parent(s):
3c8e47d
Update README.md
Browse files
README.md
CHANGED
@@ -21,8 +21,8 @@ from transformers.models.blenderbot import BlenderbotTokenizer, BlenderbotForCon
|
|
21 |
def _norm(x):
|
22 |
return ' '.join(x.strip().split())
|
23 |
|
24 |
-
tokenizer = BlenderbotTokenizer.from_pretrained('
|
25 |
-
model = BlenderbotForConditionalGeneration.from_pretrained('
|
26 |
model.eval()
|
27 |
|
28 |
utterances = [
|
|
|
21 |
def _norm(x):
|
22 |
return ' '.join(x.strip().split())
|
23 |
|
24 |
+
tokenizer = BlenderbotTokenizer.from_pretrained('thu-coai/blenderbot-1B-augesc')
|
25 |
+
model = BlenderbotForConditionalGeneration.from_pretrained('thu-coai/blenderbot-1B-augesc')
|
26 |
model.eval()
|
27 |
|
28 |
utterances = [
|