File size: 535 Bytes
5dce752 8769d0e 5dce752 8769d0e 5dce752 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from models.omnivoice_model import generate as generate_omnivoice
from models.kokoro_model import generate as generate_kokoro
from models.piper_model import generate as generate_piper
def generate_audio(model_name, sentence):
if model_name == "OmniVoice":
return generate_omnivoice(sentence)
elif model_name == "Piper":
return generate_piper(sentence)
elif model_name == "Kokoro":
return generate_kokoro(sentence)
raise ValueError(
f"Unknown model: {model_name}"
)
|