Update README.md

README.md

@@ -43,6 +43,7 @@ pip install transformers torch
 Here's a quick example to get started:
 
 ```python
+import soundfile as sf
 from transformers import AutoModel, AutoTokenizer
 
 model = AutoModel.from_pretrained("ai4bharat/vits_rasa_13", trust_remote_code=True).to("cuda")
@@ -54,6 +55,7 @@ style_id = 0  # ALEXA
 
 inputs = tokenizer(text=text, return_tensors="pt").to("cuda")
 outputs = model(inputs['input_ids'], speaker_id=speaker_id, emotion_id=style_id)
+sf.write("audio.wav", outputs.waveform.squeeze(), model.config.sampling_rate)
 print(outputs.waveform.shape)
 ```