Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 1,128 Bytes
a65e95e 5dfc565 a65e95e 12506f0 a65e95e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import { client } from '@gradio/client'
import { generateSeed } from "../utils/generateSeed.mts"
const instances: string[] = [
process.env.VC_AUDIO_GENERATION_SPACE_API_URL
]
export const generateAudio = async (prompt: string, options?: {
seed: number;
nbFrames: number;
nbSteps: number;
}) => {
const seed = options?.seed || generateSeed()
const nbFrames = options?.nbFrames || 24 // we can go up to 48 frames, but then upscaling quill require too much memory!
const nbSteps = options?.nbSteps || 35
const instance = instances.shift()
instances.push(instance)
const api = await client(instance)
const rawResponse = await api.predict('/run', [
prompt, // string in 'Prompt' Textbox component
seed, // number (numeric value between 0 and 2147483647) in 'Seed' Slider component
nbFrames, // 24 // it is the nb of frames per seconds I think?
nbSteps, // 10, (numeric value between 10 and 50) in 'Number of inference steps' Slider component
]) as any
const { name } = rawResponse?.data?.[0]?.[0] as { name: string, orig_name: string }
return `${instance}/file=${name}`
}
|