Add model images and update README

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/HELM-BERT.png filter=lfs diff=lfs merge=lfs -text
+assets/tsne_ppi_splits.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -16,7 +16,7 @@ widget:
 
 # HELM-BERT
 
-A language model for peptide representation learning using **HELM (Hierarchical Editing Language for Macromolecules)** notation.
+A peptide language model using **HELM (Hierarchical Editing Language for Macromolecules)** notation, compatible with Hugging Face Transformers.
 
 [![GitHub](https://img.shields.io/badge/GitHub-clinfo%2FHELM--BERT-black?logo=github)](https://github.com/clinfo/HELM-BERT)
 
@@ -29,6 +29,8 @@ HELM-BERT is built upon the DeBERTa architecture, designed for peptide sequences
 - **Span Masking**: Contiguous token masking with geometric distribution
 - **nGiE**: n-gram Induced Encoding layer (1D convolution, kernel size 3)
 
+<p align="center"><img src="assets/HELM-BERT.png" width="600"></p>
+
 ## Model Specifications
 
 | Parameter | Value |
@@ -82,6 +84,8 @@ Train/test 8:2, val 10% from train, 1:4 positive:negative ratio.
 - **Random**: random split
 - **aCSM**: clustering-based split on aCSM-ALL complex signatures with protein overlap pruning
 
+<p align="center"><img src="assets/tsne_ppi_splits.png" width="800"></p>
+
 ## Citation
 
 ```bibtex