import streamlit as st
from datasets import load_dataset, Audio

def app():
    st.title('Home')
    st.write('Welcome to the home page')
    ds_ckpt = st.text_input('Enter the dataset checkpoint', 'PolyAI/minds14')
    ds_lang = st.text_input('Enter the dataset language', 'en-US')
    ds_split = st.text_input('Enter the dataset split', 'train')
    # Load the dataset
    dataset = load_dataset(ds_ckpt, ds_lang, split=ds_split)
    # Display the dataset
    st.dataframe(dataset.to_pandas())

if __name__ == '__main__':
    app()