Downloads · 30 days
0
bert-torrent-nsfw
lelloman/bert-torrent-nsfw
bert-torrent-nsfw is a text classification model from lelloman. Use it when you need a label for a piece of text. The card lists the license as mit.
A BERT-tiny binary classifier for detecting NSFW content in torrent names and file listings.
Access
Public
Updated Jan 27, 2026
Repo size
17.9 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.onnx17.9 MB · 100%
At a glance
- Task
- Text Classification
- License
- mit
- Access
- Public
- Created
- Jan 27, 2026
- Updated
- Jan 27, 2026
- SHA
- f36903ba
Try a prompt
Model card
From the Hugging Face model README
BERT Torrent NSFW Classifier
A BERT-tiny binary classifier for detecting NSFW content in torrent names and file listings.
Model Details
- Architecture: BERT-tiny (2 layers, 128 hidden, 2 attention heads)
- Task: Binary classification (safe vs nsfw)
- Input: Torrent name + top 3 filenames by size
- Output: Logits for [safe, nsfw]
- Format: ONNX (embedded weights)
- Size: ~17MB
Usage
This model is designed to be used with mimmo, a Rust-based torrent content classifier.
# With mimmo CLI
mimmo --detect-nsfw "Torrent.Name.Here"
Training
Trained on a dataset of ~10k torrent names labeled as safe/nsfw using:
- BERT-tiny base model
- Binary cross-entropy loss
- AdamW optimizer
Performance
The model is used as a fallback after keyword and pattern matching stages:
- Keywords - HashSet lookup for explicit terms (confidence: 0.95)
- Patterns - Regex for JAV codes, studio patterns (confidence: 0.90)
- ML (this model) - For ambiguous cases
Limitations
- Trained primarily on English torrent names
- May not catch all NSFW content, especially with obfuscated names
- Should be used in combination with keyword/pattern stages for best results
License
MIT
- Task
- Text Classification
- License
- mit
- Languages
- en
- Created
- Jan 27, 2026
- Updated
- Jan 27, 2026