diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index d180658810..9855af3810 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -1354,6 +1354,24 @@ export const voicecraft = (model: ModelData): string[] => [ model = VoiceCraft.from_pretrained("${model.id}")`, ]; +export const vui = (model: ModelData): string[] => [ + `# !pip install git+https://github.com/fluxions-ai/vui + +import torchaudio + +from vui.inference import render +from vui.model import Vui, + +model = Vui.from_pretrained().cuda() +waveform = render( + model, + "Hey, here is some random stuff, usually something quite long as the shorter the text the less likely the model can cope!", +) +print(waveform.shape) +torchaudio.save("out.opus", waveform[0], 22050) +`, +]; + export const chattts = (): string[] => [ `import ChatTTS import torchaudio diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 867b0a0f8f..b51e19f3d6 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1063,6 +1063,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { docsUrl: "https://github.com/jasonppy/VoiceCraft", snippets: snippets.voicecraft, }, + vui: { + prettyLabel: "Vui", + repoName: "Vui", + repoUrl: "https://github.com/vui-ai/vui", + countDownloads: `path_extension:"pt"`, + snippets: snippets.vui, + }, wham: { prettyLabel: "WHAM", repoName: "wham",