Spaces:
Running
Running
Lauredecaudin
commited on
Commit
•
81910ca
1
Parent(s):
037c46a
Update pages/5-A brief note on Bias and evaluation.py
Browse files
pages/5-A brief note on Bias and evaluation.py
CHANGED
@@ -11,7 +11,7 @@ def bias_evaluation_presentation():
|
|
11 |
def toxicity_classif(text):
|
12 |
model_path = "citizenlab/distilbert-base-multilingual-cased-toxicity"
|
13 |
|
14 |
-
toxicity_classifier = pipeline("text-classification", model=model_path, tokenizer=model_path)
|
15 |
result = toxicity_classifier(text)
|
16 |
st.markdown(f"""
|
17 |
### Result
|
|
|
11 |
def toxicity_classif(text):
|
12 |
model_path = "citizenlab/distilbert-base-multilingual-cased-toxicity"
|
13 |
|
14 |
+
toxicity_classifier = pipeline("text-classification", model=model_path, tokenizer=model_path, from_pt=True)
|
15 |
result = toxicity_classifier(text)
|
16 |
st.markdown(f"""
|
17 |
### Result
|