Update README.md
Browse files
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("
|
30 |
-
model = BertForSequenceClassification.from_pretrained("
|
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."
|