Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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="
|
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, [])
|