vaibhavs10 commited on
Commit
f04e297
1 Parent(s): bf58ccb
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -5,7 +5,9 @@ import torch
5
 
6
  from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
7
 
8
- pipe = SpectrogramDiffusionPipeline.from_pretrained("google/music-spectrogram-diffusion", torch_dtype=torch.float16).to("cuda")
 
 
9
  pipe.enable_xformers_memory_efficient_attention()
10
 
11
  processor = MidiProcessor()
@@ -41,6 +43,6 @@ gr.Interface(
41
  ],
42
  title=title,
43
  description=description,
44
- theme='gradio/monochrome',
45
  examples=examples,
46
- ).launch(debug=True)
 
5
 
6
  from diffusers import SpectrogramDiffusionPipeline, MidiProcessor
7
 
8
+ pipe = SpectrogramDiffusionPipeline.from_pretrained(
9
+ "google/music-spectrogram-diffusion", torch_dtype=torch.float16
10
+ ).to("cuda")
11
  pipe.enable_xformers_memory_efficient_attention()
12
 
13
  processor = MidiProcessor()
 
43
  ],
44
  title=title,
45
  description=description,
46
+ theme="gradio/monochrome",
47
  examples=examples,
48
+ ).launch(debug=True)