MU-NLPC
/

whisper-small-audio-captioning

Inference Endpoints

Model card Files Files and versions Community

prompteus commited on May 15, 2023

Commit

567801d

•

1 Parent(s): 2b85fbf

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -60,11 +60,10 @@ Minimal example:
 ```python
 # Load model
-architecture = "openai/whisper-small"
 checkpoint = "MU-NLPC/whisper-small-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-small-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 = "..."