added a button for loading the dataset

app.py

@@ -1,16 +1,22 @@
 import streamlit as st
 from datasets import load_dataset, Audio
 
-def app():
-    st.title('Home')
-    st.write('Welcome to the home page')
+@st.cache_resource(allow_output_mutation=True)
+def load_dataset_btn():
+    st.title('Load Dataset')
     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, trust_remote_code=True)
+    dataset = load_dataset(ds_ckpt, ds_lang, split=ds_split)
     # Display the dataset
     st.dataframe(dataset.to_pandas())
 
+def app():
+    st.title('Home')
+    st.write('Welcome to the home page')
+    if st.button('Load Dataset'):
+        load_dataset_btn()
+
 if __name__ == '__main__':
     app()