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