small

app.py

@@ -4,11 +4,9 @@ import os
 
 def load_rfm_dataset(dataset_name, config_name):
     """Load the RFM dataset from HuggingFace Hub."""
-    try:
-        dataset = load_dataset(dataset_name, name=config_name)
-        return dataset, f"✅ Loaded {len(dataset)} trajectories from {dataset_name}/{config_name}"
-    except Exception as e:
-        return None, f"❌ Error loading dataset: {e}"
+
+    dataset = load_dataset(dataset_name, name=config_name)
+    return dataset, f"✅ Loaded {len(dataset)} trajectories from {dataset_name}/{config_name}"
 
 def get_available_configs(dataset_name):
     """Get available configurations for a dataset."""
@@ -127,7 +125,7 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
     
     def load_dataset(dataset_name, config_name):
         """Load the dataset and update the interface."""
-        dataset, status = load_rfm_dataset(dataset_name, name=config_name)
+        dataset, status = load_rfm_dataset(dataset_name, config_name)
         if dataset is not None:
             max_index = len(dataset) - 1
             info = f"**Dataset Info:**\n- **Total Trajectories:** {len(dataset)}\n- **Features:** {list(dataset.features.keys())}"