Update app.py

app.py

@@ -18,7 +18,7 @@ def load_huggingface_dataset(dataset_name):
     try:
         # Incrementally update progress
         progress_bar.progress(10)
-        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)
         progress_bar.progress(50)
         if hasattr(dataset, "to_pandas"):
             df = dataset.to_pandas()