Update app.py

app.py

@@ -1,21 +1,21 @@
-import gradio as gr
-from transformers import pipeline
-
-# Load audio classification pipeline
-classifier = pipeline("audio-classification", model="superb/wav2vec2-base-superb-ks")
-
-def classify_audio(audio):
-    # audio is (sample_rate, numpy array)
-    return classifier(audio)
-
-# Gradio UI
-demo = gr.Interface(
-    fn=classify_audio,
-    inputs=gr.Audio(type="filepath"),
-    outputs=gr.Label(num_top_classes=5),
-    title="Audio Classification",
-    description="Upload an audio file and classify the sound (e.g., speech, music, etc.)."
-)
-
-if __name__ == "__main__":
-    demo.launch()
+import gradio as gr
+from transformers import pipeline
+
+# Load audio classification pipeline
+classifier = pipeline("audio-classification", model="superb/wav2vec2-base-superb-ks")
+
+def classify_audio(file_path):
+    # Hugging Face audio pipeline works best with a filepath
+    return classifier(file_path)
+
+# Gradio UI
+demo = gr.Interface(
+    fn=classify_audio,
+    inputs=gr.Audio(type="filepath", label="Upload or Record Audio"),
+    outputs=gr.Label(num_top_classes=5),
+    title="Audio Classification App",
+    description="Upload or record an audio file (mp3, wav, flac, etc.) and classify the sound."
+)
+
+if __name__ == "__main__":
+    demo.launch()