Update README.md

README.md

@@ -38,7 +38,7 @@ Inference can be done using the following code (for inference code and more deta
 ```python
 import torch
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
-
+import torchaudio
 model_path = "aiola/whisper-ner-v1"
 audio_file_path = "path/to/audio/file"
 prompt = "person, company, location"  # comma separated entity tags