Update README.md

README.md

@@ -41,12 +41,12 @@ Proceed as follows:
 
 ```python
 from word2affect_polish.model_script import CustomModel # importing the custom model class
-from transformers import PreTrainedTokenizerFast
+from transformers import AutoTokenizer
 
 model_directory = "word2affect_english" # path to the cloned repository
 
 model = CustomModel.from_pretrained(model_directory)
-tokenizer = PreTrainedTokenizerFast.from_pretrained(model_directory)
+tokenizer = AutoTokenizer.from_pretrained(model_directory)
 
 inputs = tokenizer("This is a test input.", return_tensors="pt")
 outputs = model(inputs['input_ids'], inputs['attention_mask'])