Update app.py

app.py

@@ -1,10 +1,8 @@
-from transformers import pipeline, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
+from transformers import pipeline
 import gradio as gr
 
 model_id = "Tirath5504/distilhubert-finetuned-gtzan"
-processor = Wav2Vec2Processor.from_pretrained(model_id)
-model = Wav2Vec2ForSequenceClassification.from_pretrained(model_id)
-pipe = pipeline("audio-classification", model=model, processor=processor)
+pipe = pipeline("audio-classification", model=model_id)
 
 def classify_audio(filepath):
     preds = pipe(filepath)
@@ -14,6 +12,6 @@ def classify_audio(filepath):
     return outputs
 
 demo = gr.Interface(
-    fn=classify_audio, inputs=gr.Audio(type="filepath"), outputs=gr.outputs.Label()
+    fn=classify_audio, inputs=gr.Audio(type="filepath"), outputs=gr.Label(), live=True
 )
 demo.launch(debug=True)