Update app.py

app.py

@@ -8,9 +8,12 @@ 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)
+   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=pipe['audio']) 
+        scipy.io.wavfile.write(fp, rate=model.config.sampling_rate, data=output) 
         return fp.name 
 iface = gr.Interface(fn=TTS, inputs="text", outputs="audio")
 iface.launch(share=False)