Update app.py

app.py

@@ -137,7 +137,7 @@ def generate_audio_ai(text,name_model):
     #torch.cuda.empty_cache()
     with torch.no_grad():
         for chunk in _inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= speaker_embeddings,chunk_size=64):
-            yield  16000,chunk.squeeze().cpu().numpy()#.astype(np.int16).tobytes()    
+            yield  16000,chunk.cpu().numpy().squeeze()#.astype(np.int16).tobytes()    
 
     # yield generate_audio(text_answer,name_model)
 def remove_extra_spaces(text):