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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -149,9 +149,9 @@ 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
-
153
- for wav in wavs:
154
- yield (model.config.sampling_rate,wav)
155
 
156
  model_choices = gr.Dropdown(
157
  choices=[
 
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
 
156
  model_choices = gr.Dropdown(
157
  choices=[