Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
---
|
| 2 |
language: en
|
|
|
|
|
|
|
| 3 |
datasets:
|
| 4 |
- common_gen
|
| 5 |
widget:
|
|
@@ -65,7 +67,7 @@ def gen_sentence(words, max_length=32):
|
|
| 65 |
attention_mask=features['attention_mask'],
|
| 66 |
max_length=max_length)
|
| 67 |
|
| 68 |
-
return tokenizer.decode(output[0])
|
| 69 |
|
| 70 |
words = "tree plant ground hole dig"
|
| 71 |
|
|
|
|
| 1 |
---
|
| 2 |
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- common sense
|
| 5 |
datasets:
|
| 6 |
- common_gen
|
| 7 |
widget:
|
|
|
|
| 67 |
attention_mask=features['attention_mask'],
|
| 68 |
max_length=max_length)
|
| 69 |
|
| 70 |
+
return tokenizer.decode(output[0], skip_special_tokens=True)
|
| 71 |
|
| 72 |
words = "tree plant ground hole dig"
|
| 73 |
|