dance-diffusion / app.py
Fauno15's picture
Update app.py
3d49697
raw
history blame
351 Bytes
import gradio as gr
from diffusers import DiffusionPipeline
import scipy
model_id = "harmonai/maestro-150k"
pipeline = DiffusionPipeline.from_pretrained(model_id)
pipeline = pipeline.to("cuda")
def denoise(text):
audios = pipeline(audio_length_in_s=1.0).audios
return audios
gr.Interface(fn=denoise, inputs="text", outputs="audio").launch()