lauraibnz commited on
Commit
5af4730
1 Parent(s): ef58ba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 gr.make_waveform((16000, audio.audios.numpy()))
21
 
22
- demo = gr.Interface(fn=predict, inputs="prompt", outputs=gr.outputs.Audio("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.numpy())
21
 
22
+ demo = gr.Interface(fn=predict, inputs="prompt", outputs="audio")
23
  demo.launch()