Update README.md
Browse files
README.md
CHANGED
@@ -62,7 +62,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
62 |
from models import *
|
63 |
tokenizer = AutoTokenizer.from_pretrained("Hate-speech-CNERG/bert-base-uncased-hatexplain-rationale-two")
|
64 |
model = Model_Rational_Label.from_pretrained("Hate-speech-CNERG/bert-base-uncased-hatexplain-rationale-two")
|
65 |
-
inputs = tokenizer(
|
66 |
prediction_logits, _ = model(input_ids=inputs['input_ids'],attention_mask=inputs['attention_mask'])
|
67 |
```
|
68 |
|
|
|
62 |
from models import *
|
63 |
tokenizer = AutoTokenizer.from_pretrained("Hate-speech-CNERG/bert-base-uncased-hatexplain-rationale-two")
|
64 |
model = Model_Rational_Label.from_pretrained("Hate-speech-CNERG/bert-base-uncased-hatexplain-rationale-two")
|
65 |
+
inputs = tokenizer("He is a great guy", return_tensors="pt")
|
66 |
prediction_logits, _ = model(input_ids=inputs['input_ids'],attention_mask=inputs['attention_mask'])
|
67 |
```
|
68 |
|