Update app.py

app.py

@@ -5,6 +5,7 @@ from evaluate.utils import launch_gradio_widget
 import gradio as gr
 import torch
 import classify
+from whisper.model import Whisper
 from whisper.tokenizer import get_tokenizer
 from speechbrain.pretrained.interfaces import foreign_class
 from transformers import AutoModelForSequenceClassification, pipeline, WhisperTokenizer, RobertaForSequenceClassification, RobertaTokenizer, AutoTokenizer
@@ -97,7 +98,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
         model = "whisper-large"
     
         internal_lm_average_logprobs = classify.calculate_internal_lm_average_logprobs(
-            model=model,
+            model=Whisper,
             class_names=class_names,
             # class_names=classify_anxiety,
             tokenizer=tokenizer,