Update README.md

README.md

@@ -21,8 +21,14 @@ To load tokenizer from ESM, you need to install transformers with this version a
 !git clone -b add_esm-proper --single-branch https://github.com/liujas000/transformers.git 
 !pip -q install ./transformers
 
-This model is a fine-tuned version of [facebook/esm-1b](https://huggingface.co/facebook/esm-1b) on an unknown dataset.
-It achieves the following results on the evaluation set:
+This model is a fine-tuned version of [facebook/esm-1b](https://huggingface.co/facebook/esm-1b) on AAV2 dataset with ~230k sequences (Bryant et al 2020).
+
+The WT sequence (aa561-588): D E E E I R T T N P V A T E Q Y G S V S T N L Q R G N R 
+Maximum length: 50
+
+It achieves the following results on the evaluation set. 
+Note:this is result of the last epoch, I think the pushed model is loaded with best checkpoint - best val_loss, I'm not so sure though :/
+
 - Loss: 0.2250
 - Accuracy: 0.9620
 - F1: 0.9632