mskov commited on
Commit
b7b4e7f
·
1 Parent(s): 7a481f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -84,7 +84,8 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
84
  else:
85
  # model = model_cache[model_name]
86
  # class_names = classify_anxiety.split(",")
87
- class_names= class_options.get(classify_anxiety, [])
 
88
  print("class names ", class_names, "classify_anxiety ", classify_anxiety)
89
  # tokenizer = get_tokenizer(multilingual=".en" not in model_name)
90
  tokenizer= WhisperTokenizer.from_pretrained("openai/whisper-large")
 
84
  else:
85
  # model = model_cache[model_name]
86
  # class_names = classify_anxiety.split(",")
87
+ class_names_temp = class_options.get(classify_anxiety, [])
88
+ class_names = class_names_temp.split(",")
89
  print("class names ", class_names, "classify_anxiety ", classify_anxiety)
90
  # tokenizer = get_tokenizer(multilingual=".en" not in model_name)
91
  tokenizer= WhisperTokenizer.from_pretrained("openai/whisper-large")