vovahimself commited on
Commit
156d0fd
1 Parent(s): 659a5e1
Files changed (2) hide show
  1. app.py +3 -2
  2. jukwi-vqvae.ipynb +3 -2
app.py CHANGED
@@ -1,9 +1,10 @@
1
  # A simple gradio app that converts music tokens to and from audio using JukeboxVQVAE as the model and Gradio as the UI
2
 
3
- from transformers import JukeboxVQVAE
4
 
5
- import gradio as gr
6
  import torch as t
 
 
7
 
8
  model_id = 'openai/jukebox-5b-lyrics' #@param ['openai/jukebox-1b-lyrics', 'openai/jukebox-5b-lyrics']
9
 
 
1
  # A simple gradio app that converts music tokens to and from audio using JukeboxVQVAE as the model and Gradio as the UI
2
 
3
+ import sys
4
 
 
5
  import torch as t
6
+ from transformers import JukeboxVQVAE
7
+ import gradio as gr
8
 
9
  model_id = 'openai/jukebox-5b-lyrics' #@param ['openai/jukebox-1b-lyrics', 'openai/jukebox-5b-lyrics']
10
 
jukwi-vqvae.ipynb CHANGED
@@ -29,10 +29,11 @@
29
  "source": [
30
  "# A simple gradio app that converts music tokens to and from audio using JukeboxVQVAE as the model and Gradio as the UI\n",
31
  "\n",
32
- "from transformers import JukeboxVQVAE\n",
33
  "\n",
34
- "import gradio as gr\n",
35
  "import torch as t\n",
 
 
36
  "\n",
37
  "model_id = 'openai/jukebox-5b-lyrics' #@param ['openai/jukebox-1b-lyrics', 'openai/jukebox-5b-lyrics']\n",
38
  "\n",
 
29
  "source": [
30
  "# A simple gradio app that converts music tokens to and from audio using JukeboxVQVAE as the model and Gradio as the UI\n",
31
  "\n",
32
+ "import sys\n",
33
  "\n",
 
34
  "import torch as t\n",
35
+ "from transformers import JukeboxVQVAE\n",
36
+ "import gradio as gr\n",
37
  "\n",
38
  "model_id = 'openai/jukebox-5b-lyrics' #@param ['openai/jukebox-1b-lyrics', 'openai/jukebox-5b-lyrics']\n",
39
  "\n",