udit-k commited on
Commit
4232a02
·
verified ·
1 Parent(s): 5373dad

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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
- text = "Call this number to win FREE IPL FINAL tickets!!!"
36
- result = classifier(text)
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