Add cast column to speech recognition example

hezarai · Aug 15, 2023 · 6229ced · 6229ced
1 parent 8588e22
commit 6229ced
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/speech_recognition_example.py b/examples/speech_recognition_example.py
@@ -1,7 +1,8 @@
 from hezar import Model
-from datasets import load_dataset
+from datasets import load_dataset, Audio
 
 ds = load_dataset("mozilla-foundation/common_voice_11_0", "fa", split="test")
+df = ds.cast_column("audio", Audio(sampling_rate=16000))
 sample = ds[1001]
 whisper = Model.load("hezarai/whisper-small-fa")
 transcript = whisper.predict(sample["path"])  # or pass `sample["audio"]["array"]` (with the right sample rate)