Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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
|