Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import torch
-from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2Processor
+from transformers import HubertForSequenceClassification, HubertProcessor
 
 # Load the model and processor from Hugging Face
-model = Wav2Vec2ForSequenceClassification.from_pretrained("HareemFatima/distilhubert-finetuned-stutterdetection")
-processor = Wav2Vec2Processor.from_pretrained("HareemFatima/distilhubert-finetuned-stutterdetection")
+model = HubertForSequenceClassification.from_pretrained("HareemFatima/distilhubert-finetuned-stutterdetection")
+processor = HubertProcessor.from_pretrained("HareemFatima/distilhubert-finetuned-stutterdetection")
 
 # Define a function for stutter detection
 def detect_stutter(audio):