Update audio_pipe.py

audio_pipe.py

@@ -149,4 +149,4 @@ class SpeechToSpeechPipeline():
                 self.tts_task, self.tts_model, self.tts_generator, tts_sample
             )
 
-        return wav.unsqueeze(0).numpy(), sr, [text]
+        return wav, sr, [text]