ar4min commited on
Commit
1f9f282
·
verified ·
1 Parent(s): f1c5096

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -26,8 +26,8 @@ This is a fine-tuned BERT model for spam detection.
26
  from transformers import BertTokenizer, BertForSequenceClassification
27
 
28
  # Load model and tokenizer
29
- tokenizer = BertTokenizer.from_pretrained("your-username/SpamHunter")
30
- model = BertForSequenceClassification.from_pretrained("your-username/SpamHunter")
31
 
32
  # Example
33
  text = "Congratulations! You've won a $1000 gift card. Click here to claim now."
 
26
  from transformers import BertTokenizer, BertForSequenceClassification
27
 
28
  # Load model and tokenizer
29
+ tokenizer = BertTokenizer.from_pretrained("ar4min/SpamHunter")
30
+ model = BertForSequenceClassification.from_pretrained("ar4min/SpamHunter")
31
 
32
  # Example
33
  text = "Congratulations! You've won a $1000 gift card. Click here to claim now."