Update script.py

script.py

@@ -34,7 +34,7 @@ training_config = {
 
 
 # WRAP YOUR DATASET CURATION FUNCTIONS IN THIS FUNCTION
-def prepare_dataset(name="Voxel51/Data-Centric-Visual-AI-Challenge-Train-Set"):
+def prepare_dataset():
     """
     Prepare the dataset for model training. 
     
@@ -53,7 +53,7 @@ def prepare_dataset(name="Voxel51/Data-Centric-Visual-AI-Challenge-Train-Set"):
     """
     
     # DO NOT MODIFY THIS LINE
-    dataset = fouh.load_from_hub(name, split="train")
+    dataset = fouh.load_from_hub("/tmp/data/train")
     
     # WRAP YOUR DATA CURATION FUNCTIONS HERE
     dataset = shuffle_data(dataset)