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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -85,7 +85,11 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
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")
 
85
  # model = model_cache[model_name]
86
  # class_names = classify_anxiety.split(",")
87
  class_names_temp = class_options.get(classify_anxiety, [])
88
+ class_str = ""
89
+ for elm in class_names_temp:
90
+ class_str += elm + ","
91
+ #class_names = class_names_temp.split(",")
92
+ class_names = class_str.split(",")
93
  print("class names ", class_names, "classify_anxiety ", classify_anxiety)
94
  # tokenizer = get_tokenizer(multilingual=".en" not in model_name)
95
  tokenizer= WhisperTokenizer.from_pretrained("openai/whisper-large")