Update app.py

app.py

@@ -58,7 +58,7 @@ if input_option == "Use Hugging Face Dataset":
     if st.button("Load Dataset"):
         try:
             with st.spinner("Loading dataset..."):
-                dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True)
+                dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True, uniform_split=True)
                 st.session_state.df = pd.DataFrame(dataset)
                 st.success(f"Dataset '{dataset_name}' loaded successfully!")
                 st.dataframe(st.session_state.df.head())