Update app.py

app.py

@@ -9,6 +9,8 @@ from models import SynthesizerTrn
 from text.symbols import symbols
 from text import text_to_sequence
 
+import IPython.display as ipd
+
 import json
 import math
 
@@ -36,8 +38,8 @@ def vc_fn(input):
         x_tst = stn_tst.unsqueeze(0)
         x_tst_lengths = torch.LongTensor([stn_tst.size(0)])
         audio = net_g.infer(x_tst, x_tst_lengths, noise_scale=.667, noise_scale_w=0.8, length_scale=1)[0][0,0].data.cpu().float().numpy()
-    sampling_rate = 22050
-    return (sampling_rate, audio)
+    #sampling_rate = 22050
+    return audio
         
 app = gr.Blocks()
 with app:
@@ -45,7 +47,8 @@ with app:
         with gr.TabItem("Basic"):
             vc_input = gr.Textbox(label="Input Message")
             vc_submit = gr.Button("Convert", variant="primary")
-            vc_output = gr.Audio(label="Output Audio")
+            #vc_output = gr.Audio(label="Output Audio")
+            vc_output = ipd.display(ipd.Audio(audio, rate=hps.data.sampling_rate))
         vc_submit.click(vc_fn, [ vc_input], [vc_output])
 
     app.launch()