Update README.md

README.md

@@ -43,6 +43,7 @@ General-purpose GBT model trained on ESM2 embeddings.
 # How to Use
 
 Download Model from Hugging Face
+
 from huggingface_hub import hf_hub_download
 
 # Download ACE2 RBD model/General model
@@ -50,19 +51,26 @@ from huggingface_hub import hf_hub_download
 model_path = hf_hub_download(repo_id="hbp5181/ESM2-GBT", filename="ACE2_RBD_ESM2-GBT.json")
 
 Load Model in CatBoost
+
 from catboost import CatBoostRegressor
 
 model = CatBoostRegressor()
+
 model.load_model(model_path, format="json")
 
-# Dataset
-Training Details
+
+# Training Details
 
 •	Feature Extraction: ESM2 embeddings (33-layer transformer, 650M params)
+
 •	Training Algorithm: CatBoost Gradient Boosting
+
 •	Dataset: 
+
       ACE2 RBD: https://github.com/jbloomlab/SARSr-CoV_homolog_survey
+      
       General: https://zenodo.org/records/14271435
+      
 •	Evaluation Metrics: RMSE, R^2
 
 # Applications