wasmdashai commited on
Commit
3d972f3
·
verified ·
1 Parent(s): a558fe1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ def modelspeech(text=TXT,name_model="wasmdashai/vits-ar-sa-huba-v2",speaking_r
149
  wavs=_inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= None,is_streaming=True)
150
  # with torch.no_grad():
151
  # wav = model(input_ids=inputs["input_ids"].cuda()).waveform.cpu().numpy().reshape(-1)#.detach()
152
- return (model.config.sampling_rate,list(wavs))
153
  # for wav in wavs:
154
  # yield (model.config.sampling_rate,wav)
155
 
 
149
  wavs=_inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= None,is_streaming=True)
150
  # with torch.no_grad():
151
  # wav = model(input_ids=inputs["input_ids"].cuda()).waveform.cpu().numpy().reshape(-1)#.detach()
152
+ return (model.config.sampling_rate,np.concatenate(list(wavs)))
153
  # for wav in wavs:
154
  # yield (model.config.sampling_rate,wav)
155