Hugging Face's logo

Spaces:
fffiloni
/
spectrogram-to-music
Sleeping

App Files Community
105
fffiloni commited on
Commit
be62a3b
·
1 Parent(s): 1c42a58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ MODEL_ID = "riffusion/riffusion-model-v1"
12
  pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
13
  pipe = pipe.to("cuda")
14
  pipe2 = StableDiffusionImg2ImgPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
15
- pipe2 = pipe2.to(device)
16
 
17
  spectro_from_wav = gr.Interface.load("spaces/fffiloni/audio-to-spectrogram")
18
 
 
12
  pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
13
  pipe = pipe.to("cuda")
14
  pipe2 = StableDiffusionImg2ImgPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
15
+ pipe2 = pipe2.to("cuda")
16
 
17
  spectro_from_wav = gr.Interface.load("spaces/fffiloni/audio-to-spectrogram")
18