Update app.py

app.py

@@ -8,7 +8,7 @@ from IPython.display import Audio
 # Import the model
 from jukebox.models import MODELS
 
-model = jukebox.make_vqvae(MODELS.5B_LYRICS)
+model = jukebox.make_vqvae(MODELS['5B_LYRICS'], device="cpu")
 
 
 # Generate music