mskov commited on
Commit
789ba90
·
1 Parent(s): 78affd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def classify_toxicity(audio_file, text_input, selected_model):
27
 
28
  # Load the selected toxicity classification model
29
  #toxicity_module = evaluate.load("toxicity", selected_model)
30
- toxicity_module = evaluate.load("toxicity", 'DaNLP/da-electra-hatespeech-detection', module_type="measurement", toxic_label={"not offensive": 0, "offensive": 1})
31
 
32
  toxicity_results = toxicity_module.compute(predictions=[transcribed_text])
33
 
 
27
 
28
  # Load the selected toxicity classification model
29
  #toxicity_module = evaluate.load("toxicity", selected_model)
30
+ toxicity_module = evaluate.load("toxicity", 'DaNLP/da-electra-hatespeech-detection', module_type="measurement")
31
 
32
  toxicity_results = toxicity_module.compute(predictions=[transcribed_text])
33