Update README.md

README.md

@@ -15,11 +15,14 @@ More information needed
 
 ## Training and evaluation data
 
+!pip install setfit  
+
 from datasets import load_dataset  
 import numpy as np  
 from sklearn.metrics import accuracy_score  
 from keras.preprocessing.text import Tokenizer  
 from keras.utils import pad_sequences  
+import tensorflow as tf  
 from huggingface_hub import from_pretrained_keras  
   
 sst2 = load_dataset("SetFit/sst2")  
@@ -36,7 +39,9 @@ pred=reloaded_model.predict(test_padded)
 pred_bin = np.argmax(pred,1)  
 accuracy_score(pred_bin, sst2['test']['label'])
 
-0.9176276771004942
+0.9176276771004942  
+
+reloaded_model.summary()
 
 ## Training procedure