Automatic speech recognition
In this exercise, you use AI to transcribe audio into text automatically! You'll be working with the VCTK Corpus again, which includes around 44-hours of speech uttered by English speakers with various accents. You'll use OpenAI's Whisper tiny model, which contains only 37M parameters to preprocess the VCTK audio data and generate the corresponding text.
The audio preprocessor (processor) has been loaded, as has the WhisperForConditionalGeneration module. A sample audio datapoint (sample) has already been loaded.
This exercise is part of the course
Multi-Modal Models with Hugging Face
Exercise instructions
- Load the
WhisperForConditionalGenerationpretrained model using theopenai/whisper-tinycheckpoint. - Preprocess the
sampledatapoint with the required sampling rate of16000. - Generate the tokens from the model using the
.input_featuresattribute of the preprocessed inputs.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Load the pretrained model
model = ____
model.config.forced_decoder_ids=None
# Preprocess the sample audio
input_preprocessed = ____(____, sampling_rate=____, return_tensors="pt", return_attention_mask=True)
# Generate the IDs of the recognized tokens
predicted_ids = ____
transcription = processor.decode(predicted_ids[0], skip_special_tokens=True)
print(transcription)