Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,7 @@ As in the original plT5 implementation, the training dataset was tokenized into
|
|
80 |
```python
|
81 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
82 |
|
83 |
-
|
84 |
tokenizer = T5Tokenizer.from_pretrained("Voicelab/vlt5-base-keywords")
|
85 |
|
86 |
task_prefix = "Keywords: "
|
|
|
80 |
```python
|
81 |
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
82 |
|
83 |
+
model = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
|
84 |
tokenizer = T5Tokenizer.from_pretrained("Voicelab/vlt5-base-keywords")
|
85 |
|
86 |
task_prefix = "Keywords: "
|