Spaces:
Paused
Paused
Update app.py
Browse files
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,
|
| 40 |
-
|
|
|
|
| 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
|