teticio commited on
Commit
2b5d8ed
1 Parent(s): abb41ea

add ddim option to apps

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  2. streamlit_app.py +4 -2
app.py CHANGED
@@ -19,13 +19,15 @@ demo = gr.Interface(fn=generate_spectrogram_audio_and_loop,
19
  title="Audio Diffusion",
20
  description="Generate audio using Huggingface diffusers.\
21
  This takes about 20 minutes without a GPU, so why not make yourself a \
22
- cup of tea in the meantime?",
 
23
  inputs=[
24
  gr.Dropdown(label="Model",
25
  choices=[
26
  "teticio/audio-diffusion-256",
27
  "teticio/audio-diffusion-breaks-256",
28
- "teticio/audio-diffusion-instrumental-hiphop-256"
 
29
  ],
30
  value="teticio/audio-diffusion-256")
31
  ],
 
19
  title="Audio Diffusion",
20
  description="Generate audio using Huggingface diffusers.\
21
  This takes about 20 minutes without a GPU, so why not make yourself a \
22
+ cup of tea in the meantime? (Or try the teticio/audio-diffusion-ddim-256 \
23
+ model which is faster.)",
24
  inputs=[
25
  gr.Dropdown(label="Model",
26
  choices=[
27
  "teticio/audio-diffusion-256",
28
  "teticio/audio-diffusion-breaks-256",
29
+ "teticio/audio-diffusion-instrumental-hiphop-256",
30
+ "teticio/audio-diffusion-ddim-256"
31
  ],
32
  value="teticio/audio-diffusion-256")
33
  ],
streamlit_app.py CHANGED
@@ -10,11 +10,13 @@ if __name__ == "__main__":
10
  st.header("Audio Diffusion")
11
  st.markdown("Generate audio using Huggingface diffusers.\
12
  This takes about 20 minutes without a GPU, so why not make yourself a \
13
- cup of tea in the meantime?")
 
14
 
15
  model_id = st.selectbox("Model", [
16
  "teticio/audio-diffusion-256", "teticio/audio-diffusion-breaks-256",
17
- "teticio/audio-diffusion-instrumental-hiphop-256"
 
18
  ])
19
  audio_diffusion = AudioDiffusion(model_id=model_id)
20
 
 
10
  st.header("Audio Diffusion")
11
  st.markdown("Generate audio using Huggingface diffusers.\
12
  This takes about 20 minutes without a GPU, so why not make yourself a \
13
+ cup of tea in the meantime? (Or try the teticio/audio-diffusion-ddim-256 \
14
+ model which is faster.)")
15
 
16
  model_id = st.selectbox("Model", [
17
  "teticio/audio-diffusion-256", "teticio/audio-diffusion-breaks-256",
18
+ "teticio/audio-diffusion-instrumental-hiphop-256",
19
+ "teticio/audio-diffusion-ddim-256"
20
  ])
21
  audio_diffusion = AudioDiffusion(model_id=model_id)
22