Update app.py

app.py

@@ -5,7 +5,7 @@ from evaluate.utils import launch_gradio_widget
 import gradio as gr
 import torch
 from speechbrain.pretrained.interfaces import foreign_class
-from transformers import AutoModelForSequenceClassification, WhisperForConditionalGeneration, pipeline, RobertaForSequenceClassification, RobertaTokenizer, AutoTokenizer
+from transformers import AutoModelForSequenceClassification, pipeline, RobertaForSequenceClassification, RobertaTokenizer, AutoTokenizer
 # pull in emotion detection
 # --- Add element for specification
 # pull in text classification
@@ -39,7 +39,7 @@ def classify_toxicity(audio_file, text_input, classify_anxiety):
         # Extract the transcribed text
         # transcribed_text = transcription_results["transcription"]
         '''
-        model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
+        model = whisper.load_model("base")
         transcribed_text = model.transcribe(audio_file)
         
         #### Emotion classification ####