File size: 478 Bytes
180d0f0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# DistilBERT for Text Classification
This repository contains a fine-tuned DistilBERT model for text classification. The model is designed to classify text into four categories: SAFE, JAILBREAK, INJECTION, and PHISHING.
## Model Details
- Base model: DistilBERT (distilbert-base-uncased)
- Task: Sequence Classification
- Number of labels: 4
- Labels: SAFE, JAILBREAK, INJECTION, PHISHING
## Usage
To use this model, you can leverage the Hugging Face Transformers library:
|