teticio commited on
Commit
600e950
1 Parent(s): 760dafb

added instrumental hiphop

Browse files
Files changed (3) hide show
  1. app.py +2 -1
  2. notebooks/test_model.ipynb +11 -9
  3. streamlit_app.py +4 -3
app.py CHANGED
@@ -24,7 +24,8 @@ demo = gr.Interface(fn=generate_spectrogram_audio_and_loop,
24
  gr.Dropdown(label="Model",
25
  choices=[
26
  "teticio/audio-diffusion-256",
27
- "teticio/audio-diffusion-breaks-256"
 
28
  ],
29
  value="teticio/audio-diffusion-256")
30
  ],
 
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
  ],
notebooks/test_model.ipynb CHANGED
@@ -10,7 +10,7 @@
10
  },
11
  {
12
  "cell_type": "code",
13
- "execution_count": 1,
14
  "id": "6c7800a6",
15
  "metadata": {},
16
  "outputs": [],
@@ -27,7 +27,7 @@
27
  },
28
  {
29
  "cell_type": "code",
30
- "execution_count": 2,
31
  "id": "b447e2c4",
32
  "metadata": {},
33
  "outputs": [],
@@ -39,7 +39,7 @@
39
  },
40
  {
41
  "cell_type": "code",
42
- "execution_count": 3,
43
  "id": "c2fc0e7a",
44
  "metadata": {},
45
  "outputs": [],
@@ -61,16 +61,18 @@
61
  },
62
  {
63
  "cell_type": "code",
64
- "execution_count": null,
65
  "id": "97f24046",
66
  "metadata": {},
67
  "outputs": [],
68
  "source": [
69
- "#@markdown teticio/audio-diffusion-256 - trained on my Spotify \"liked\" playlist\n",
 
 
70
  "\n",
71
- "#@markdown teticio/audio-diffusion-256-breaks - trained on samples used in music\n",
72
  "\n",
73
- "model_id = \"teticio/audio-diffusion-256\" #@param [\"teticio/audio-diffusion-256\", \"teticio/audio-diffusion-256-breaks\"]"
74
  ]
75
  },
76
  {
@@ -83,7 +85,7 @@
83
  },
84
  {
85
  "cell_type": "code",
86
- "execution_count": 4,
87
  "id": "a3d45c36",
88
  "metadata": {},
89
  "outputs": [],
@@ -119,7 +121,7 @@
119
  },
120
  {
121
  "cell_type": "code",
122
- "execution_count": 12,
123
  "id": "f028a3c8",
124
  "metadata": {},
125
  "outputs": [],
 
10
  },
11
  {
12
  "cell_type": "code",
13
+ "execution_count": 4,
14
  "id": "6c7800a6",
15
  "metadata": {},
16
  "outputs": [],
 
27
  },
28
  {
29
  "cell_type": "code",
30
+ "execution_count": 5,
31
  "id": "b447e2c4",
32
  "metadata": {},
33
  "outputs": [],
 
39
  },
40
  {
41
  "cell_type": "code",
42
+ "execution_count": 6,
43
  "id": "c2fc0e7a",
44
  "metadata": {},
45
  "outputs": [],
 
61
  },
62
  {
63
  "cell_type": "code",
64
+ "execution_count": 7,
65
  "id": "97f24046",
66
  "metadata": {},
67
  "outputs": [],
68
  "source": [
69
+ "#@markdown teticio/audio-diffusion-256 - trained on my Spotify \"liked\" playlist\n",
70
+ "\n",
71
+ "#@markdown teticio/audio-diffusion-breaks-256 - trained on samples used in music\n",
72
  "\n",
73
+ "#@markdown teticio/audio-diffusion-instrumental-hiphop-256 - trained on instrumental hiphop\n",
74
  "\n",
75
+ "model_id = \"teticio/audio-diffusion-256\" #@param [\"teticio/audio-diffusion-256\", \"teticio/audio-diffusion-breaks-256\", \"audio-diffusion-instrumenal-hiphop-256\"]"
76
  ]
77
  },
78
  {
 
85
  },
86
  {
87
  "cell_type": "code",
88
+ "execution_count": null,
89
  "id": "a3d45c36",
90
  "metadata": {},
91
  "outputs": [],
 
121
  },
122
  {
123
  "cell_type": "code",
124
+ "execution_count": null,
125
  "id": "f028a3c8",
126
  "metadata": {},
127
  "outputs": [],
streamlit_app.py CHANGED
@@ -12,9 +12,10 @@ if __name__ == "__main__":
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(
16
- "Model",
17
- ["teticio/audio-diffusion-256", "teticio/audio-diffusion-breaks-256"])
 
18
  audio_diffusion = AudioDiffusion(model_id=model_id)
19
 
20
  if st.button("Generate"):
 
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
 
21
  if st.button("Generate"):