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 |
|