YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
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:
- Downloads last month
- 16