MU-NLPC
/

whisper-large-v2-audio-captioning

Inference Endpoints

Model card Files Files and versions Community

prompteus commited on May 15, 2023

Commit

4cad71c

•

1 Parent(s): e61873f

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -58,11 +58,10 @@ Minimal example:
 ```python
 # Load model
-architecture = "openai/whisper-large-v2"
 checkpoint = "MU-NLPC/whisper-large-v2-audio-captioning"
-model = audiocap.WhisperForAudioCaptioning.from_pretrained(checkpoint)
 tokenizer = transformers.WhisperTokenizer.from_pretrained(checkpoint, language="en", task="transcribe")
-feature_extractor = transformers.WhisperFeatureExtractor.from_pretrained(architecture)
 # Load and preprocess audio
 input_file = "..."

 ```python
 # Load model
 checkpoint = "MU-NLPC/whisper-large-v2-audio-captioning"
+model = WhisperForAudioCaptioning.from_pretrained(checkpoint)
 tokenizer = transformers.WhisperTokenizer.from_pretrained(checkpoint, language="en", task="transcribe")
+feature_extractor = transformers.WhisperFeatureExtractor.from_pretrained(checkpoint)
 # Load and preprocess audio
 input_file = "..."