# audio_classification_app.py
import gradio as gr
from transformers import pipeline

classifier = pipeline("audio-classification", model="superb/wav2vec2-base-superb-ks")

def classify_audio(audio_file):
    return classifier(audio_file)

demo = gr.Interface(
    fn=classify_audio,
    inputs=gr.Audio(type="filepath"),
    outputs=gr.Label(num_top_classes=5),
    title="Audio Classification",
    description="Upload or record audio to classify (keywords dataset model)."
)

if __name__ == "__main__":
    demo.launch()