Instructions to use holistic-ai/bias_classifier_albertv2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use holistic-ai/bias_classifier_albertv2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="holistic-ai/bias_classifier_albertv2")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("holistic-ai/bias_classifier_albertv2") model = AutoModelForSequenceClassification.from_pretrained("holistic-ai/bias_classifier_albertv2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
|
@@ -36,7 +36,7 @@
|
|
| 36 |
"1": "Stereotype"
|
| 37 |
},
|
| 38 |
"label2id": {
|
| 39 |
-
"
|
| 40 |
-
"
|
| 41 |
}
|
| 42 |
}
|
|
|
|
| 36 |
"1": "Stereotype"
|
| 37 |
},
|
| 38 |
"label2id": {
|
| 39 |
+
"Non-Stereotype": 0,
|
| 40 |
+
"Stereotype": 1
|
| 41 |
}
|
| 42 |
}
|