Update app.py

app.py

@@ -83,6 +83,8 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
         return toxicity_score, classification_output, emo_dict[text_lab[0]], transcribed_text
         # return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"
     else: 
+        model = whisper.load_model("large")
+        model_cache[model_name] = model
         # model = model_cache[model_name]
         # class_names = classify_anxiety.split(",")
         class_names_list = class_options.get(classify_anxiety, [])
@@ -98,7 +100,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
         model = "whisper-large"
     
         internal_lm_average_logprobs = classify.calculate_internal_lm_average_logprobs(
-            model=Whisper,
+            model=model,
             class_names=class_names,
             # class_names=classify_anxiety,
             tokenizer=tokenizer,