import gradio as gr
import numpy as np
from datasets import load_dataset

minds = load_dataset("PolyAI/minds14", name="en-AU", split="train")
id2label = minds.features["intent_class"].int2str


def generate_audio():
    example = minds.shuffle()[0]
    audio = example["audio"]
    return (
        audio["sampling_rate"],
        audio["array"],
    ), id2label(example["intent_class"])


with gr.Blocks() as demo:
    with gr.Column():
        for _ in range(4):
            audio, label = generate_audio()
            output = gr.Audio(audio, label=label)

demo.launch()