Views
No views yet
LABEL_0 output indicates non-racist textLABEL_1 output indicates racist text1from transformers import pipeline
2
3RACISM_MODEL = "davidmasip/racism"
4racism_analysis_pipe = pipeline("text-classification",
5 model=RACISM_MODEL, tokenizer=RACISM_MODEL)
6
7results = racism_analysis_pipe("Unos menas agreden a una mujer.")
8
9
10def clean_labels(results):
11 for result in results:
12 label = "Non-racist" if results["label"] == "LABEL_0" else "Racist"
13 result["label"] = label
14
15
16clean_labels(results)
17print(results)