---

import torch
from TTS.api import TTS

# Модель сама скачается при первом запуске
MODEL_NAME = "tts_models/multilingual/multi-dataset/xtts_v2"

device = "cuda" if torch.cuda.is_available() else "cpu"
print("Device:", device)

tts = TTS(MODEL_NAME).to(device)

text = "Привет. Это тестовая озвучка на русском языке голосом из референса."

reference_wav = "reference.wav"   # путь к вашему референсу
output_wav = "output.wav"

tts.tts_to_file(
    text=text,
    speaker_wav=reference_wav,
    language="ru",
    file_path=output_wav
)

print("Saved:", output_wav)

from IPython.display import Audio

Audio(output_wav)

tts.tts_to_file(
    text="Привет. Это тест с несколькими референсами.",
    speaker_wav=[
        "ref1.wav",
        "ref2.wav",
        "ref3.wav",
    ],
    language="ru",
    file_path="output_multi_ref.wav"
)

Audio("output_multi_ref.wav")