Hugging Face's logo

Spaces:
harmonai
/
dance-diffusion
Runtime error

App Files Community
14
dance-diffusion
File size: 2,168 Bytes 
6374438
3d49697
97bb0e0
6374438
 
97bb0e0
 
 
77f3a64
97bb0e0
 
77f3a64
4a0f7d0
5af5097
97bb0e0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1ee5d9d
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, 60.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)