update

app.py

@@ -1,18 +1,18 @@
 # adding a comment
 import gradio as gr
-from datasets import load_dataset
+from datasets import load_dataset as load_dataset_hf
 import os
 
 def load_rfm_dataset(dataset_name, config_name):
     """Load the RFM dataset from HuggingFace Hub."""
 
-    dataset = load_dataset(dataset_name, name=config_name)
+    dataset = load_dataset_hf(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."""
     try:
-        dataset_info = load_dataset(dataset_name, trust_remote_code=True)
+        dataset_info = load_dataset_hf(dataset_name, trust_remote_code=True)
         configs = list(dataset_info.keys())
         return configs
     except Exception as e: