From a45110ee4f04750cf43214b591847cffa1074310 Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Wed, 23 Oct 2024 20:39:54 -0700 Subject: [PATCH] Update docstring (#1954) AudioConverter is registered as "keras_hub.layers.WhisperAudioConverter" and not as part of models. --- keras_hub/src/models/whisper/whisper_audio_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_hub/src/models/whisper/whisper_audio_converter.py b/keras_hub/src/models/whisper/whisper_audio_converter.py index 633042f547..9890109bac 100644 --- a/keras_hub/src/models/whisper/whisper_audio_converter.py +++ b/keras_hub/src/models/whisper/whisper_audio_converter.py @@ -39,7 +39,7 @@ class WhisperAudioConverter(AudioConverter): audio_tensor = tf.ones((8000,), dtype="float32") # Compute the log-mel spectrogram. - audio_converter = keras_hub.models.WhisperAudioConverter.from_preset( + audio_converter = keras_hub.layers.WhisperAudioConverter.from_preset( "whisper_base_en", ) audio_converter(audio_tensor)