Update app.py

app.py

@@ -2,7 +2,8 @@ import streamlit as st
 from transformers import pipeline, AutoTokenizer, AutoModelForTextToWaveform
 
 # Load the audio classification model
-audio_classification_model = pipeline("audio-classification", model="HareemFatima/distilhubert-finetuned-stutterdetection")
+processor = AutoProcessor.from_pretrained("HareemFatima/distilhubert-finetuned-stutterdetection")
+model = AutoModelForAudioClassification.from_pretrained("HareemFatima/distilhubert-finetuned-stutterdetection")
 
 # Load the TTS tokenizer and model
 tts_tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")