Update README.md

README.md

@@ -38,6 +38,25 @@ vuln-cat is a classification model based on fine-tuning of scibert. It categoriz
     'xxe'
 ]
 ```
+
+## Usage
+```python
+from transformers import pipeline
+
+text = 'A path traversal exists in a specific dll of Trend Micro Mobile Security (Enterprise) 9.8 SP5 which could allow an authenticated remote attacker to delete arbitrary files.'
+
+classifier = pipeline(
+    "text-classification",
+    model="conflick0/vuln-cat",
+    padding=True,
+    truncation=True,
+    max_length=512,
+)
+
+classifier(text)
+# [{'label': 'directory_traversal', 'score': 0.9969494938850403}]
+```
+
 ## Intended uses & limitations
 
 More information needed