MU-NLPC
/

whisper-tiny-audio-captioning

Inference Endpoints

Model card Files Files and versions Community

prompteus commited on May 15, 2023

Commit

8b93dee

•

1 Parent(s): ebe8e1e

Update README.md

Files changed (1) hide show

README.md +3 -4

README.md CHANGED Viewed

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