Update README.md

README.md

@@ -38,6 +38,7 @@ You can use the models through Huggingface's Transformers library. Check our Git
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 import torch
+import torchaudio
 import re
 from string import Template
 prompt_template = Template("Human: ${inst} </s> Assistant: ")
@@ -79,7 +80,7 @@ abc_pattern = r'(X:\d+\n(?:[^\n]*\n)+)'
 abc_notation = re.findall(abc_pattern, response+'\n')[0]
 s = Score.from_abc(abc_notation)
 audio = Synthesizer().render(s, stereo=True)
-dump_wav('cm_music_piece.wav', audio, sample_rate=44100, use_int16=True)
+torchaudio.save('cm_music_piece.wav', torch.FloatTensor(audio), 44100)
 ```
 
 ## Chat demo