Update README.md

README.md

@@ -7,12 +7,13 @@ metrics:
 model-index:
 - name: my_awesome_model
   results: []
+pipeline_tag: text-classification
 ---
 
 <!-- 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. -->
 
-# my_awesome_model
+# vuln-cat
 
 This model is a fine-tuned version of [allenai/scibert_scivocab_uncased](https://huggingface.co/allenai/scibert_scivocab_uncased) on the None dataset.
 It achieves the following results on the evaluation set:
@@ -21,8 +22,22 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
-
+vuln-cat is a classification model based on fine-tuning of scibert. It categorizes CVE summaries into 11 types of vulnerabilities, with class labels including:
+```
+[
+    'csrf',
+    'directory_traversal',
+    'file_inclusion', 
+    'input_validation', 
+    'memory_corruption', 
+    'open_redirect', 
+    'overflow', 
+    'sql_injection', 
+    'ssrf',
+    'xss', 
+    'xxe'
+]
+```
 ## Intended uses & limitations
 
 More information needed
@@ -63,4 +78,4 @@ The following hyperparameters were used during training:
 - Transformers 4.38.2
 - Pytorch 2.2.1+cu121
 - Datasets 2.18.0
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2