Update README.md

README.md

@@ -111,7 +111,7 @@ predicted_token_class_ids = logits.argmax(-1)
 # Note that tokens are classified rather then input words which means that
 # there might be more predicted token classes than words.
 # Multiple token classes might account for the same word
-predicted_tokens_classes = [model_tuned.config.id2label[t.item()] for t in predicted_token_class_ids[0]]
+predicted_tokens_classes = [model.config.id2label[t.item()] for t in predicted_token_class_ids[0]]
 predicted_tokens_classes
 ```