mskov commited on
Commit
40948af
1 Parent(s): 4b9eea9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,8 +59,8 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
59
 
60
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
61
 
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, [])
 
59
 
60
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
61
 
62
+ text_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
63
+
64
  sequence_to_classify = transcribed_text
65
  print(classify_anxiety, class_options)
66
  candidate_labels = class_options.get(classify_anxiety, [])