Update README.md

README.md

@@ -21,7 +21,7 @@ And then we suggest loading it like follows:
 
 ```
 import torch
-from transformers import T5EncoderModel
+from transformers import T5EncoderModel, AutoTokenizer
 from transformers.modeling_outputs import SequenceClassifierOutput
 
 class T5EncoderRerank(torch.nn.Module):
@@ -51,4 +51,5 @@ model = T5EncoderRerank(original_model,bf16=True)
 model.load_state_dict(torch.load(path_checkpoint,map_location=torch.device("cpu")))
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 model.to(device)
+tokenizer = AutoTokenizer.from_pretrained(original_model)
 ```