mskov commited on
Commit
e95ab8a
1 Parent(s): 335e90e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def classify_toxicity(audio_file, text_input):
16
  transcribed_text = text_input
17
 
18
  # Load the selected toxicity classification model
19
- toxicity_module = evaluate.load("toxicity", "mskov/roberta-base-toxicity")
20
  #toxicity_module = evaluate.load("toxicity", 'DaNLP/da-electra-hatespeech-detection', module_type="measurement")
21
 
22
  toxicity_results = toxicity_module.compute(predictions=[transcribed_text])
 
16
  transcribed_text = text_input
17
 
18
  # Load the selected toxicity classification model
19
+ toxicity_module = evaluate.load("toxicity", "facebook/roberta-hate-speech-dynabench-r4-target")
20
  #toxicity_module = evaluate.load("toxicity", 'DaNLP/da-electra-hatespeech-detection', module_type="measurement")
21
 
22
  toxicity_results = toxicity_module.compute(predictions=[transcribed_text])