Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
21 |
|
22 |
-
demo = gr.Interface(fn=predict, inputs="prompt", outputs=
|
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()
|