Update app.py

app.py

@@ -1,12 +1,16 @@
+import tempfile ,os
 import gradio as gr
 from transformers import VitsModel, AutoTokenizer,pipeline
 import torch
+import scipy
 
 
 def TTS(text):
    model = VitsModel.from_pretrained("SeyedAli/Persian-Speech-synthesis")
    tokenizer = AutoTokenizer.from_pretrained("SeyedAli/Persian-Speech-synthesis")
-   pipe = pipeline("text-to-speech", model=model,tokenizer=tokenizer) 
-   return pipe(text)
-iface = gr.Interface(fn=TTS, inputs="text", outputs="text")
+   pipe = pipeline("text-to-speech", model=model,tokenizer=tokenizer)
+   with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
+        scipy.io.wavfile.write(fp, rate=model.config.sampling_rate, data=pipe['audio']) 
+        return fp.name 
+iface = gr.Interface(fn=TTS, inputs="text", outputs="audio")
 iface.launch(share=False)