AgaMiko commited on
Commit
d0f8cce
1 Parent(s): 4bad107

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- vlt5 = T5ForConditionalGeneration.from_pretrained("Voicelab/vlt5-base-keywords")
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: "