Update app.py

app.py

@@ -1,8 +1,10 @@
-from transformers import pipeline
+from transformers import pipeline, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
 import gradio as gr
 
 model_id = "Tirath5504/distilhubert-finetuned-gtzan"
-pipe = pipeline("audio-classification", model=model_id)
+processor = Wav2Vec2Processor.from_pretrained(model_id)
+model = Wav2Vec2ForSequenceClassification.from_pretrained(model_id)
+pipe = pipeline("audio-classification", model=model, processor=processor)
 
 def classify_audio(filepath):
     preds = pipe(filepath)