Update app.py

app.py

@@ -10,11 +10,9 @@ def TTS(text):
    model = VitsModel.from_pretrained("SeyedAli/Persian-Speech-synthesis")
    tokenizer = AutoTokenizer.from_pretrained("SeyedAli/Persian-Speech-synthesis")
    inputs = tokenizer(text, return_tensors="pt")
-   with torch.no_grad():
-        output = model(**inputs).waveform
-   # with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
-   #      scipy.io.wavfile.write(fp, rate=model.config.sampling_rate, data=output) 
-   #      return fp.name
-       return output
+   pipe = pipeline("text-to-speech", model=model,tokenizer=tokenizer)
+   with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
+        scipy.io.wavfile.write(fp, rate=pipe(text)['sampling_rate'], data=pipe(text)['audio']) 
+        return fp.name
 iface = gr.Interface(fn=TTS, inputs="text", outputs="text")
 iface.launch(share=False)