update

app.py

@@ -5,9 +5,11 @@ import os
 
 def load_rfm_dataset(dataset_name, config_name):
     """Load the RFM dataset from HuggingFace Hub."""
-
-    dataset = load_dataset_hf(dataset_name, name=config_name)
-    return dataset, f"✅ Loaded {len(dataset)} trajectories from {dataset_name}/{config_name}"
+    try:
+        dataset = load_dataset_hf(dataset_name, name=config_name, split="train")
+        return dataset, f"✅ Loaded {len(dataset)} trajectories from {dataset_name}/{config_name}"
+    except Exception as e:
+        return None, f"❌ Error loading dataset: {e}"
 
 def get_available_configs(dataset_name):
     """Get available configurations for a dataset."""
@@ -16,6 +18,7 @@ def get_available_configs(dataset_name):
         configs = list(dataset_info.keys())
         return configs
     except Exception as e:
+        print(f"Error getting configs: {e}")
         return []
 
 def visualize_trajectory(dataset, index):