Update README.md

README.md

@@ -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)
+model = WhisperForAudioCaptioning.from_pretrained(checkpoint)
 tokenizer = transformers.WhisperTokenizer.from_pretrained(checkpoint, language="en", task="transcribe")
-feature_extractor = transformers.WhisperFeatureExtractor.from_pretrained(architecture)
+feature_extractor = transformers.WhisperFeatureExtractor.from_pretrained(checkpoint)
 
 # Load and preprocess audio
 input_file = "..."