Update README.md

README.md

@@ -1,20 +1,19 @@
 
 # Hate Speech Classification Model
 
-This model is fine-tuned for Named Entity Recognition (NER) specifically for intent detection.
+This model is fine-tuned for hatespeech detection using pre annotated NER intent tags.
 
 ## Model Details
 
 - Model Type: RoBERTa
-- Number of Labels: 6
-- Labels: ['O', 'derogation', 'animosity', 'threatening', 'hatecrime', 'comparison']
+- Number of Labels: 2
 
 ## Usage
 
 ```python
-from transformers import AutoModelForTokenClassification, AutoTokenizer
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
-model = AutoModelForTokenClassification.from_pretrained("afcarvallo/hate-speech-intent-model-ihc")
+model = AutoModelForSequenceClassification.from_pretrained("afcarvallo/hate-speech-intent-model-ihc")
 tokenizer = AutoTokenizer.from_pretrained("afcarvallo/hate-speech-intent-model-ihc")
 
 ```