Improve model card: Add pipeline tag, paper, project, code links, and descriptive tags

README.md

@@ -1,25 +1,32 @@
 ---
+base_model: distilbert-base-uncased
+datasets:
+- CIRCL/vulnerability-scores
 library_name: transformers
 license: apache-2.0
-base_model: distilbert-base-uncased
-tags:
-- generated_from_trainer
 metrics:
 - accuracy
+pipeline_tag: text-classification
+tags:
+- generated_from_trainer
+- security
+- vulnerability
+- classification
+- distilbert
 model-index:
 - name: vulnerability-severity-classification-distilbert-base-uncased
   results: []
-datasets:
-- CIRCL/vulnerability-scores
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
 # vulnerability-severity-classification-distilbert-base-uncased
 
 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)
+
 It achieves the following results on the evaluation set:
 - Loss: 0.6447
 - Accuracy: 0.7595