Update app.py

app.py

@@ -5,6 +5,7 @@ import re
 import time
 import uuid
 from io import StringIO
+from io import BytesIO
 
 import gradio as gr
 import spaces
@@ -169,10 +170,18 @@ def predict(
         keep_len = calculate_keep_len(prompt, language)
         out["wav"] = out["wav"][:keep_len]
 
-        print(out)
-        print(out["wav"])
+        # print(out)
+        # print(out["wav"])
 
-        torchaudio.save("output.wav", torch.tensor(out["wav"]).unsqueeze(0), 24000)
+        buffer = BytesIO()
+
+        # torchaudio.save("output.wav", torch.tensor(out["wav"]).unsqueeze(0), 24000)
+        torchaudio.save(buffer, torch.tensor(out["wav"]).unsqueeze(0), 24000, format='wav')
+
+        buffer.seek(0)
+        wav_base64 = base64.b64encode(buffer.read()).decode('utf-8')
+        print(wav_base64)
+        wav_base64
 
     except RuntimeError as e:
         if "device-side assert" in str(e):
@@ -321,4 +330,4 @@ with gr.Blocks(analytics_enabled=False) as demo:
     )
 
 demo.queue()
-demo.launch(debug=True, show_api=True, share=True)
+demo.launch(debug=True, show_api=True, share=True)