Update README.md

README.md

@@ -10,6 +10,19 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassifica
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 identity_model = AutoModelForSequenceClassification.from_pretrained("Mridul2003/identity-hate-detector").to(device)
 identity_tokenizer = AutoTokenizer.from_pretrained("Mridul2003/identity-hate-detector")
+identity_inputs = identity_tokenizer(final_text, return_tensors="pt", padding=True, truncation=True)
+    if 'token_type_ids' in identity_inputs:
+        del identity_inputs['token_type_ids']
+    identity_inputs = {k: v.to(device) for k, v in identity_inputs.items()}
+    with torch.no_grad():
+        identity_outputs = identity_model(**identity_inputs)
+    identity_probs = torch.sigmoid(identity_outputs.logits)
+    identity_prob = identity_probs[0][1].item()
+    not_identity_prob = identity_probs[0][0].item()
+
+    results["identity_hate_custom"] = identity_prob
+    results["not_identity_hate_custom"] = not_identity_prob
+
 ```
 
 # Offensive Language Classifier (Fine-Tuned on Custom Dataset)