Update README.md

README.md

@@ -55,12 +55,14 @@ from catboost import CatBoostRegressor
 model = CatBoostRegressor()
 model.load_model(model_path, format="json")
 
-# Predictions using your own dataset!
+# Dataset
 Training Details
 
 •	Feature Extraction: ESM2 embeddings (33-layer transformer, 650M params)
 •	Training Algorithm: CatBoost Gradient Boosting
-•	Dataset: your own dataset
+•	Dataset: 
+      ACE2 RBD: https://github.com/jbloomlab/SARSr-CoV_homolog_survey
+      General: https://zenodo.org/records/14271435
 •	Evaluation Metrics: RMSE, R^2
 
 # Applications