Update app.py

app.py

@@ -16,4 +16,9 @@ controlnet = ControlNetModel.from_pretrained("lauraibnz/midi-audioldm", torch_dt
 pipe = AudioLDMControlNetPipeline.from_pretrained("cvssp/audioldm-m-full", controlnet=controlnet, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
-audio = pipe("techno", audio_length_in_s=10, num_inference_steps=20, midi=midi, controlnet_conditioning_scale=1.0)
+def predict(prompt, midi, audio_length_in_s=10, num_inference_steps=20, controlnet_conditioning_scale=1.0):
+    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)
+    return audio
+
+demo = gr.Interface(fn=predict, inputs="prompt", outputs="audio")
+demo.launch()