Views
No views yet
1# Chạy FastAPI server
2python api_server.py
3
4# Test bằng curl
5curl -X POST "http://localhost:8000/filter" \
6 -H "Content-Type: application/json" \
7 -d '{"text": "Sản phẩm tốt", "strict_mode": true}'1from transformers import AutoTokenizer, AutoModel
2import torch
3
4# Load model
5model = AutoModel.from_pretrained("your-username/vn-comment-filter-kids")
6tokenizer = AutoTokenizer.from_pretrained("your-username/vn-comment-filter-kids")
7
8# Inference
9text = "Comment cần kiểm tra"
10inputs = tokenizer(text, return_tensors="pt", max_length=256, padding=True, truncation=True)
11
12with torch.no_grad():
13 outputs = model(**inputs)
14 prediction = torch.argmax(outputs.logits, dim=-1).item()
15
16print("CLEAN" if prediction == 0 else "TOXIC").
├── model.safetensors # Model weights
├── config.json # Model config
├── tokenizer files # Tokenizer artifacts
├── preprocessing_config.json # Text preprocessing rules
├── specialist_navigator.pkl # Logistic model cho OFF vs HATE
├── specialist_power.pkl # Logistic model cho toxicity prob
├── requirements.txt # Python dependencies
└── README.md # Documentation này