Staticaliza commited on
Commit
f44a081
·
verified ·
1 Parent(s): c1a557d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -35,9 +35,14 @@ def tts(text,lang,speaker,
35
  model.generate(model.prepare_conditioning(cond),
36
  cfg_scale=float(cfg),sampling_params=dict(min_p=float(minp)))
37
  )[0].cpu().clamp_(-1,1).numpy()
 
 
 
 
38
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
39
- sf.write(tmp.name, wav, model.autoencoder.sampling_rate,
40
- subtype="PCM_16")
 
41
  return tmp.name
42
 
43
  langs=supported_language_codes
 
35
  model.generate(model.prepare_conditioning(cond),
36
  cfg_scale=float(cfg),sampling_params=dict(min_p=float(minp)))
37
  )[0].cpu().clamp_(-1,1).numpy()
38
+ wav = np.squeeze(wav) # drop singleton channel
39
+ if wav.ndim == 2: # (C, N) ➜ (N, C)
40
+ wav = wav.T
41
+
42
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
43
+ sf.write(tmp.name, wav,
44
+ model.autoencoder.sampling_rate,
45
+ format="WAV", subtype="PCM_16")
46
  return tmp.name
47
 
48
  langs=supported_language_codes