Update tasks/audio.py

tasks/audio.py

@@ -136,10 +136,10 @@ def extract_embedding(audio_example):
     return {"embedding": embeddings.numpy()}
 
 # Apply embedding extraction to training data
-train_embeddings = train_test["train"].map(extract_embedding)
+train_embeddings = dataset["train"].map(extract_embedding)
 
 # Apply embedding extraction to testing data
-test_embeddings = train_test["test"].map(extract_embedding)
+test_embeddings = dataset["test"].map(extract_embedding)
 
 X_train, y_train = [], []
 X_test, y_test = [], []