Create TTS/tts.py

TTS/tts.py

@@ -0,0 +1,14 @@
+from transformers import VitsModel, AutoTokenizer
+import torch
+
+
+model = VitsModel.from_pretrained("facebook/mms-tts-ara")
+tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara")
+
+def text_to_speech(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    with torch.no_grad():
+        output = model(**inputs)
+    waveform = output.waveform.squeeze().numpy()
+    sampling_rate = model.config.sampling_rate
+    return waveform, sampling_rate