hashest commited on
Commit
e1fdbe1
·
1 Parent(s): fddaaf5

modify audio tokenizer example

Browse files
Files changed (1) hide show
  1. sparktts/models/audio_tokenizer.py +2 -2
sparktts/models/audio_tokenizer.py CHANGED
@@ -155,9 +155,9 @@ if __name__ == "__main__":
155
  model_dir="pretrained_models/Spark-TTS-0.5B",
156
  device=device,
157
  )
158
- wav_path = "local/zhisheng_prompt.wav"
159
 
160
  global_tokens, semantic_tokens = tokenizer.tokenize(wav_path)
161
 
162
  wav_rec = tokenizer.detokenize(global_tokens.squeeze(0), semantic_tokens)
163
- sf.write("local/wav_rec.wav", wav_rec, 16000)
 
155
  model_dir="pretrained_models/Spark-TTS-0.5B",
156
  device=device,
157
  )
158
+ wav_path = "example/prompt_audio.wav"
159
 
160
  global_tokens, semantic_tokens = tokenizer.tokenize(wav_path)
161
 
162
  wav_rec = tokenizer.detokenize(global_tokens.squeeze(0), semantic_tokens)
163
+ sf.write("example/prompt_recon.wav", wav_rec, 16000)