Update README.md
Browse files
README.md
CHANGED
|
@@ -32,12 +32,12 @@ tokenizer = BertTokenizer.from_pretrained("udit-k/HamSpamBERT")
|
|
| 32 |
model = BertForSequenceClassification.from_pretrained("udit-k/HamSpamBERT")
|
| 33 |
|
| 34 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
print(result)
|
| 38 |
```
|
| 39 |
```
|
| 40 |
[{'label': 'LABEL_1', 'score': 0.9999189376831055}]
|
|
|
|
| 41 |
```
|
| 42 |
|
| 43 |
## Model description
|
|
|
|
| 32 |
model = BertForSequenceClassification.from_pretrained("udit-k/HamSpamBERT")
|
| 33 |
|
| 34 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 35 |
+
print(classifier("Call this number to win FREE IPL FINAL tickets!!!"))
|
| 36 |
+
print(classifier("Call me when you reach home :)"))
|
|
|
|
| 37 |
```
|
| 38 |
```
|
| 39 |
[{'label': 'LABEL_1', 'score': 0.9999189376831055}]
|
| 40 |
+
[{'label': 'LABEL_0', 'score': 0.9999370574951172}]
|
| 41 |
```
|
| 42 |
|
| 43 |
## Model description
|