Update README.md

README.md

@@ -16,8 +16,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 # protBERTbfd_AAV2_classification
 
-This model is a fine-tuned version of [Rostlab/prot_bert_bfd](https://huggingface.co/Rostlab/prot_bert_bfd) on an unknown dataset.
-It achieves the following results on the evaluation set:
+This model is a fine-tuned version of [Rostlab/prot_bert_bfd](https://huggingface.co/Rostlab/prot_bert_bfd) 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.1341
 - Accuracy: 0.9615
 - F1: 0.9627