Spaces:
Runtime error
Runtime error
File size: 2,167 Bytes
6374438 3d49697 97bb0e0 6374438 97bb0e0 77f3a64 97bb0e0 77f3a64 4a0f7d0 5af5097 97bb0e0 80dd0b9 97bb0e0 3d49697 97bb0e0 |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import gradio as gr
from diffusers import DiffusionPipeline
import scipy.io.wavfile
def load_model(model_id):
pipeline = DiffusionPipeline.from_pretrained(model_id)
pipeline = pipeline.to("cuda")
return pipeline
def denoise(length_sec,model):
pipeline = load_model(model)
audios = pipeline(audio_length_in_s=length_sec).audios
for audio in audios:
scipy.io.wavfile.write("test.wav", pipeline.unet.sample_rate, audio.transpose())
return "test.wav"
block = gr.Blocks()
with block:
gr.HTML(
"""
<div style="text-align: center; max-width: 700px; margin: 0 auto;">
<div
style="
display: inline-flex;
align-items: center;
gap: 0.8rem;
font-size: 1.75rem;
"
>
<h1 style="font-weight: 900; margin-bottom: 7px;">
Dance Diffusion
</h1>
</div>
<p style="margin-bottom: 10px; font-size: 94%">
Dance Diffusion is the first in a suite of generative audio tools for producers and musicians to be released by Harmonai
</p>
</div>
"""
)
with gr.Group():
with gr.Box():
length = gr.Slider(1.0, 6.0, value=3.0, step=0.5, label="Audio length in seconds")
model = gr.Dropdown(choices=["harmonai/maestro-150k", "harmonai/jmann-small-190k", "harmonai/honk-140k", "harmonai/unlocked-250k","harmonai/jmann-large-580k","harmonai/glitch-440k"], value="harmonai/maestro-150k",type="value", label="Model")
out = gr.Audio(label="Output", type="filepath")
btn = gr.Button("Submit").style(full_width=True)
btn.click(denoise, inputs=[length,model], outputs=out)
gr.HTML('''
<div class="footer" style="text-align: center; max-width: 700px; margin: 0 auto;">
<p>Model by <a href="https://huggingface.co/harmonai" style="text-decoration: underline;" target="_blank">Harmonai</a>
</p>
</div>
''')
block.launch(debug=True) |