Spam Detector (Fine-Tuned BERT)
Usage
from transformers import BertTokenizer, BertForSequenceClassification
tokenizer = BertTokenizer.from_pretrained("Olivier-52/BERT_spam_detector")
model = BertForSequenceClassification.from_pretrained("Olivier-52/BERT_spam_detector")
inputs = tokenizer("Your text here", return_tensors="pt")
outputs = model(**inputs)
- Downloads last month
- 14
Model tree for Olivier-52/BERT_spam_detector
Base model
google-bert/bert-base-uncased