mskov commited on
Commit
efc6979
·
1 Parent(s): e385084

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -159,11 +159,10 @@ def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, expli
159
  )
160
  average_logprobs -= internal_lm_average_logprobs
161
  scores = average_logprobs.softmax(-1).tolist()
162
- return {class_name: score for class_name, score in zip(class_names, scores)}
163
  if toxicity_score > threshold:
164
  print("threshold exceeded!! Launch intervention")
165
  miso_label_dict = {label: score for label, score in classify_anxiety[0].items()}
166
- holder1 = ""
167
  holder2 = ""
168
  holder3= " "
169
  return holder1, miso_label_dict, holder2, holder3
 
159
  )
160
  average_logprobs -= internal_lm_average_logprobs
161
  scores = average_logprobs.softmax(-1).tolist()
162
+ holder1 = {class_name: score for class_name, score in zip(class_names, scores)}
163
  if toxicity_score > threshold:
164
  print("threshold exceeded!! Launch intervention")
165
  miso_label_dict = {label: score for label, score in classify_anxiety[0].items()}
 
166
  holder2 = ""
167
  holder3= " "
168
  return holder1, miso_label_dict, holder2, holder3