Update README.md

README.md

@@ -20,6 +20,10 @@ The full dataset contains 11,970 protein sequences, including Cellwall (87), Cyt
 The highly imbalanced sample sizes across the six categories in this dataset pose a significant challenge for classification.  
 
 The dataset was downloaded from the website at [**DeepLocPro - 1.0**](https://services.healthtech.dtu.dk/services/DeepLocPro-1.0/). 
+
+# Model training code at GitHub
+[**PPPSL-ESM2**](https://github.com/pengsihua2023/PPPSL-ESM2)
+
 # How to use **PPPSL-ESM2**
 ### An example
 Pytorch and transformers libraries should be installed in your system.