Create data_loader.py

data_loader.py

@@ -0,0 +1,10 @@
+def load_hf_dataset():
+    dataset = load_dataset("your_username/dataset_name",
+        split={
+            "train": "train",
+            "validation": "validation"
+        },
+        token="YOUR_HF_TOKEN"
+    )
+    return dataset
+