Update README.md

README.md

@@ -115,21 +115,22 @@ from transformers import AutoTokenizer, AutoModel, TFAutoModel
 import numpy as np
 
 MODEL = "cardiffnlp/twitter-roberta-base"
+tokenizer = AutoTokenizer.from_pretrained(MODEL)
+
 text = "Good night 😊"
 text = preprocess(text)
-tokenizer = AutoTokenizer.from_pretrained(MODEL)
 
 # Pytorch
-encoded_input = tokenizer(text, return_tensors='pt')
 model = AutoModel.from_pretrained(MODEL)
+encoded_input = tokenizer(text, return_tensors='pt')
 features = model(**encoded_input)
 features = features[0].detach().cpu().numpy() 
 features_mean = np.mean(features[0], axis=0) 
 #features_max = np.max(features[0], axis=0)
 
 # # Tensorflow
-# encoded_input = tokenizer(text, return_tensors='tf')
 # model = TFAutoModel.from_pretrained(MODEL)
+# encoded_input = tokenizer(text, return_tensors='tf')
 # features = model(encoded_input)
 # features = features[0].numpy()
 # features_mean = np.mean(features[0], axis=0)