mskov commited on
Commit
9c9e849
1 Parent(s): 6666837

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -62,6 +62,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
62
  text_classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli")
63
 
64
  sequence_to_classify = transcribed_text
 
65
  candidate_labels = class_options.get(classify_anxiety, [])
66
  # classification_output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
67
  classification_output = text_classifier(sequence_to_classify, candidate_labels, multi_label=True)
 
62
  text_classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli")
63
 
64
  sequence_to_classify = transcribed_text
65
+ print(classify_anxiety, class_options)
66
  candidate_labels = class_options.get(classify_anxiety, [])
67
  # classification_output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
68
  classification_output = text_classifier(sequence_to_classify, candidate_labels, multi_label=True)