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}`
}