Update README.md

README.md

@@ -13,6 +13,16 @@ The dataset and models are available here: https://github.com/punyajoy/HateXplai
 
 Please use the **Model_Rational_Label** class inside [models.py](models.py) to load the models. The default prediction in this hosted inference API may be wrong due to the use of different class initialisations.
 
+~~~
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+### from models.py
+from models import *
+tokenizer = AutoTokenizer.from_pretrained("Hate-speech-CNERG/bert-base-uncased-hatexplain-rationale-two")
+model = Model_Rational_Label.from_pretrained("Hate-speech-CNERG/bert-base-uncased-hatexplain-rationale-two")
+inputs = tokenizer('He is a great guy", return_tensors="pt")
+prediction_logits, _ = model(input_ids=inputs['input_ids'],attention_mask=inputs['attention_mask'])
+~~~
+
 **For more details about our paper**
 
 Binny Mathew, Punyajoy Saha, Seid Muhie Yimam, Chris Biemann, Pawan Goyal, and Animesh Mukherjee "[HateXplain: A Benchmark Dataset for Explainable Hate Speech Detection)". Accepted at AAAI 2021.