Update app.py

app.py

@@ -7,6 +7,7 @@ os.system("pip install numpy==1.22.0")
 from modelscope.models.audio.tts import SambertHifigan
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
+import IPython.display as ipd
 
 model_dir = os.path.abspath("./pretrain_work_dir")
 
@@ -35,7 +36,8 @@ inference = pipeline(task=Tasks.text_to_speech, model=model_id)
 
 def infer(text):
     output = inference(input=text)
-    return output
+    audio = ipd.Audio(output["output_wav"], rate=16000)
+    return audio
 
 app = gr.Blocks()