Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import urllib.request
 from pydub import AudioSegment
+import numpy as np
 
 def function1(prompt):
     response = requests.post("https://tommy24-testing3.hf.space/run/predict", json={
@@ -19,9 +20,9 @@ def function1(prompt):
     data = "https://matthijs-speecht5-tts-demo.hf.space/file="+data
     file_name, headers = urllib.request.urlretrieve(data, "speech.wav")
     sound = AudioSegment.from_wav(file_name)
-    mp3_file = "speech.mp3"
-    sound.export(mp3_file, format="mp3")
-    return data
+    sound.export("speech.mp3", format="mp3")
+    audio_array = np.array(sound.get_array_of_samples())
+    return audio_array
 
-iface = gr.Interface(fn=function1, inputs="text", outputs=[gr.Audio(label="Generated Speech", type="numpy")])
+iface = gr.Interface(fn=function1, inputs="text", outputs=gr.Audio(label="Audio", type="numpy"))
 iface.launch()