Spaces:

JoshKeesee
/

Riffusion-Musiccaps

Running

JoshKeesee commited on Feb 16

Commit

31eea77

verified ·

1 Parent(s): 3ca1c59

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,11 +1,11 @@
 import gradio as gr
 import torch
 from spectro import wav_bytes_from_spectrogram_image
-from diffusers import StableDiffusionPipeline
 device = "cpu"
 MODEL_ID = "Hyeon2/riffusion-musiccaps"
-pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID)
 pipe = pipe.to(device)
 def predict(prompt):

 import gradio as gr
 import torch
 from spectro import wav_bytes_from_spectrogram_image
+from diffusers import DiffusionPipeline
 device = "cpu"
 MODEL_ID = "Hyeon2/riffusion-musiccaps"
+pipe = DiffusionPipeline.from_pretrained(MODEL_ID)
 pipe = pipe.to(device)
 def predict(prompt):