def prepare_dataset(example):
     audio = example["audio"]

     example.update(processor(audio=audio["array"], text=example["text"], sampling_rate=16000))
     return example

Apply the prepare_dataset function to a sample:

prepare_dataset(lj_speech[0])

The processor has now added input_values and labels, and the sampling rate has also been correctly downsampled to 16kHz. You can pass your processed dataset to the model now!