lauraibnz commited on
Commit
a43d1de
1 Parent(s): 6f55f2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ pipe = pipe.to(device)
17
  def predict(prompt, midi_file="test.mid", audio_length_in_s=10, num_inference_steps=20, controlnet_conditioning_scale=1.0):
18
  midi = PrettyMIDI(midi_file)
19
  audio = pipe(prompt, midi=midi, audio_length_in_s=audio_length_in_s, num_inference_steps=num_inference_steps, controlnet_conditioning_scale=controlnet_conditioning_scale)
20
- return (16000, audio.audios.numpy())
21
 
22
  demo = gr.Interface(fn=predict, inputs="text", outputs="audio")
23
  demo.launch()
 
17
  def predict(prompt, midi_file="test.mid", audio_length_in_s=10, num_inference_steps=20, controlnet_conditioning_scale=1.0):
18
  midi = PrettyMIDI(midi_file)
19
  audio = pipe(prompt, midi=midi, audio_length_in_s=audio_length_in_s, num_inference_steps=num_inference_steps, controlnet_conditioning_scale=controlnet_conditioning_scale)
20
+ return (16000, audio.audios)
21
 
22
  demo = gr.Interface(fn=predict, inputs="text", outputs="audio")
23
  demo.launch()