Update app.py

app.py

@@ -83,7 +83,9 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
         # return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"
     else: 
         # model = model_cache[model_name]
-        class_names = classify_anxiety.split(",")
+        # class_names = classify_anxiety.split(",")
+        class_names= class_options.get(classify_anxiety, [])
+        print("class names ", class_names, "classify_anxiety ", classify_anxiety)
         # tokenizer = get_tokenizer(multilingual=".en" not in model_name)
         tokenizer= WhisperTokenizer.from_pretrained("openai/whisper-large")
         model = "whisper-large"