Update README

README.md

@@ -18,6 +18,8 @@ widget:
 
 A language model for peptide representation learning using **HELM (Hierarchical Editing Language for Macromolecules)** notation.
 
+[![GitHub](https://img.shields.io/badge/GitHub-clinfo%2FHELM--BERT-black?logo=github)](https://github.com/clinfo/HELM-BERT)
+
 ## Model Description
 
 HELM-BERT is built upon the DeBERTa architecture, designed for peptide sequences in HELM notation:
@@ -27,8 +29,6 @@ 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)
 
-Please check the [official repository](https://github.com/clinfo/HELM-BERT) for more implementation details and updates.
-
 ## Model Specifications
 
 | Parameter | Value |