Update README.md

README.md

@@ -22,10 +22,8 @@ model-index:
 
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the dataset [CIRCL/vulnerability-scores](https://huggingface.co/datasets/CIRCL/vulnerability-scores).
 
-This model is part of the work presented in the paper [VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification](https://huggingface.co/papers/2507.03607).
-
 **Project Page**: [https://vulnerability.circl.lu](https://vulnerability.circl.lu)
-**Code Repository**: [https://github.com/vulnerability-lookup/ML-Gateway](https://github.com/vulnerability-lookup/ML-Gateway)
+**Code Repository**: [https://github.com/vulnerability-lookup/VulnTrain](https://github.com/vulnerability-lookup/VulnTrain)
 
 It achieves the following results on the evaluation set:
 - Loss: 0.6447