Update audio_pipe.py

audio_pipe.py

@@ -4,6 +4,7 @@ from pathlib import Path
 from typing import List, Tuple
 import tempfile
 import soundfile as sf
+import gradio as gr
 
 import numpy as np
 import torch
@@ -150,12 +151,11 @@ class SpeechToSpeechPipeline():
             wav, sr = TTSHubInterface.get_prediction(
                 self.tts_task, self.tts_model, self.tts_generator, tts_sample
             )
-        temp_name = ""
+        temp_file = ""
         with tempfile.NamedTemporaryFile(suffix=".wav") as tmp_output_file:
             sf.write(tmp_output_file, wav.detach().cpu().numpy(), sr)
             tmp_output_file.seek(0)
-            temp_name = tmp_output_file.name
-            print(temp_name)
+            temp_file = gr.Audio(tmp_output_file.name)
 
         # return wav, sr, [text]
-        return temp_name
+        return temp_file