Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
-import urllib.request
 import requests
-# from pydub import AudioSegment
+import urllib.request
+from pydub import AudioSegment
 
 def function1(prompt):
     response = requests.post("https://matthijs-speecht5-tts-demo.hf.space/run/predict", json={
@@ -12,8 +12,11 @@ def function1(prompt):
     }).json()
     data = response["data"][0]["name"]
     data = "https://matthijs-speecht5-tts-demo.hf.space/file="+data
-    speech = urllib.request.urlretrieve(data, "speech.mp3")
-    return 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 mp3_file
 
 iface = gr.Interface(fn=function1, inputs="text", outputs="text")
-iface.launch()
+iface.launch()