changed to train set

prepare.py

@@ -13,7 +13,7 @@ if __name__ == "__main__":
         print("Dataset loaded from cache.")
     else:
         # Load dataset using datasets.load_dataset()
-        ds = datasets.load_dataset("renumics/speech_commands_enriched", split="test")
+        ds = datasets.load_dataset("renumics/speech_commands_enriched", split="train")
         print("Dataset loaded using datasets.load_dataset().")
 
         df = ds.to_pandas()