Spaces:
Runtime error
Runtime error
Rename voice_infer_xtts.py to bark_infer_cpu.py
Browse files- bark_infer_cpu.py +22 -0
- voice_infer_xtts.py +0 -22
bark_infer_cpu.py
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from bark import generate_audio, preload_models
|
| 2 |
+
import numpy as np
|
| 3 |
+
import scipy.io.wavfile
|
| 4 |
+
import argparse
|
| 5 |
+
|
| 6 |
+
parser = argparse.ArgumentParser()
|
| 7 |
+
parser.add_argument("--text_file", required=True)
|
| 8 |
+
parser.add_argument("--output", default="bark_output.wav")
|
| 9 |
+
args = parser.parse_args()
|
| 10 |
+
|
| 11 |
+
# Read text
|
| 12 |
+
with open(args.text_file, "r") as f:
|
| 13 |
+
text = f.read()
|
| 14 |
+
|
| 15 |
+
# Load Bark CPU models
|
| 16 |
+
preload_models(use_gpu=False)
|
| 17 |
+
|
| 18 |
+
# Generate audio
|
| 19 |
+
audio_array = generate_audio(text, history_prompt="v2/en_speaker_9")
|
| 20 |
+
|
| 21 |
+
# Save output
|
| 22 |
+
scipy.io.wavfile.write(args.output, rate=22050, data=audio_array)
|
voice_infer_xtts.py
DELETED
|
@@ -1,22 +0,0 @@
|
|
| 1 |
-
# voice_infer_xtts.py
|
| 2 |
-
|
| 3 |
-
import torch
|
| 4 |
-
from TTS.api import TTS
|
| 5 |
-
import argparse
|
| 6 |
-
|
| 7 |
-
parser = argparse.ArgumentParser()
|
| 8 |
-
parser.add_argument("--text_file", required=True)
|
| 9 |
-
parser.add_argument("--output", default="xtts_output.wav")
|
| 10 |
-
parser.add_argument("--language", default="hi") # 'hi' for Hindi, 'en' for English
|
| 11 |
-
parser.add_argument("--voice", default="female") # Optionally switch voice profiles
|
| 12 |
-
args = parser.parse_args()
|
| 13 |
-
|
| 14 |
-
# Read text
|
| 15 |
-
with open(args.text_file, "r") as f:
|
| 16 |
-
text = f.read()
|
| 17 |
-
|
| 18 |
-
# Load Coqui XTTS
|
| 19 |
-
tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2", gpu=True)
|
| 20 |
-
|
| 21 |
-
# Generate voice
|
| 22 |
-
tts.tts_to_file(text=text, file_path=args.output, language=args.language, speaker=args.voice)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|