From 1d1dab091a30dae99961cb5b2ec392609676bd9d Mon Sep 17 00:00:00 2001
From: Sam Khoze <68170403+SamKhoze@users.noreply.github.com>
Date: Fri, 14 Jun 2024 20:58:14 +0530
Subject: [PATCH] Add files via upload

---
 tts_generation.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 tts_generation.py

diff --git a/tts_generation.py b/tts_generation.py
new file mode 100644
index 0000000..d991e72
--- /dev/null
+++ b/tts_generation.py
@@ -0,0 +1,27 @@
+import os
+import torch
+import argparse
+from TTS.api import TTS
+
+def main():
+    parser = argparse.ArgumentParser(description="Run TTS with specified parameters.")
+    
+    parser.add_argument('--model', type=str, default="tts_models/multilingual/multi-dataset/xtts_v2", help="The TTS model to use.")
+    parser.add_argument('--text', type=str, required=True, help="The text to be converted to speech.")
+    parser.add_argument('--speaker_wav', type=str, required=True, help="The path to the speaker's wav file for voice cloning.")
+    parser.add_argument('--language', type=str, default="en", help="The language of the text.")
+    parser.add_argument('--output_file', type=str, default="output.wav", help="The output file path for the synthesized speech.")
+    parser.add_argument('--device', type=str, choices=["cpu", "mps","cuda"], default="cpu" if torch.cuda.is_available() else "cpu", help="The device to run the model on.")
+    
+    args = parser.parse_args()
+
+    # Init TTS
+    tts = TTS()
+    tts.load_tts_model_by_path(model_path=args.model,config_path=os.path.join(args.model,"config.json"))
+    # tts.to(args.device)
+
+    # Run TTS and save to file
+    tts.tts_to_file(text=args.text, speaker_wav=args.speaker_wav, language=args.language, file_path=args.output_file)
+
+if __name__ == "__main__":
+    main()