Fix for the fma_dataset variable (#59)

kahrendt · Feb 18, 2025 · 4b3116a · 4b3116a
1 parent f0e9857
commit 4b3116a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/notebooks/basic_training_notebook.ipynb b/notebooks/basic_training_notebook.ipynb
@@ -185,9 +185,9 @@
     "\n",
     "    # Save clips to 16-bit PCM wav files\n",
     "    fma_dataset = datasets.Dataset.from_dict({\"audio\": [str(i) for i in Path(\"fma/fma_small\").glob(\"**/*.mp3\")]})\n",
-    "    audioset_dataset = audioset_dataset.cast_column(\"audio\", datasets.Audio(sampling_rate=16000))\n",
-    "    for row in tqdm(audioset_dataset):\n",
-    "        name = row['audio']['path'].split('/')[-1].replace(\".flac\", \".wav\")\n",
+    "    fma_dataset = fma_dataset.cast_column(\"audio\", datasets.Audio(sampling_rate=16000))\n",
+    "    for row in tqdm(fma_dataset):\n",
+    "        name = row['audio']['path'].split('/')[-1].replace(\".mp3\", \".wav\")\n",
     "        scipy.io.wavfile.write(os.path.join(output_dir, name), 16000, (row['audio']['array']*32767).astype(np.int16))\n"
    ]
   },