Downloads · 30 days
11
1% of all-time downloads
tiktok-text-safety-classifier
KhoiBui/tiktok-text-safety-classifier
tiktok-text-safety-classifier is a text classification model from KhoiBui. Use it when you need a label for a piece of text. The card lists the license as mit.
This model classifies TikTok video text content (captions, comments) as safe or harmful.
All-time downloads
1.2K
Public
Parameters
560M
2.3 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors2.2 GB · 99%
At a glance
- Task
- Text Classification
- License
- mit
- Model type
- xlm-roberta
- Access
- Public
- Created
- Jan 28, 2026
- Updated
- Jan 28, 2026
- SHA
- f2ca7f70
Datasets
Model card
From the Hugging Face model README
TikTok Content Safety - Text Classifier
This model classifies TikTok video text content (captions, comments) as safe or harmful.
Usage
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("KhoiBui/tiktok-text-safety-classifier")
model = AutoModelForSequenceClassification.from_pretrained("KhoiBui/tiktok-text-safety-classifier")
text = "your text here"
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
outputs = model(**inputs)
prediction = outputs.logits.argmax(-1).item() # 0=safe, 1=harmful
Model Details
- Base Model: XLM-RoBERTa / CafeBERT
- Task: Binary classification (safe/harmful)
- Training Data: TikTok video metadata (Vietnamese + English)
- Task
- Text Classification
- Type
- xlm-roberta
- License
- mit
- Languages
- vi, en
- Created
- Jan 28, 2026
- Updated
- Jan 28, 2026