Commit
•
e4867ce
1
Parent(s):
2250d38
Update README.md
Browse files
README.md
CHANGED
|
@@ -69,4 +69,12 @@ outputs = model.generate(
|
|
| 69 |
)
|
| 70 |
|
| 71 |
tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 72 |
```
|
|
|
|
| 69 |
)
|
| 70 |
|
| 71 |
tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
| 72 |
+
```
|
| 73 |
+
which outputs the following top-5 predictions (using contrained search)
|
| 74 |
+
```
|
| 75 |
+
['Albert Einstein',
|
| 76 |
+
'Erwin Schrödinger',
|
| 77 |
+
'Werner Bruschke',
|
| 78 |
+
'Werner von Habsburg',
|
| 79 |
+
'Werner von Moltke']
|
| 80 |
```
|