Update app.py

app.py

@@ -6,8 +6,11 @@ import os
 from IPython.display import Audio
 
 # Import the model
+from jukebox.models import MODELS
+
 model = jukebox.make_vqvae(MODELS.5B_LYRICS)
 
+
 # Generate music
 def generate_music(temperature=1.0, top_k=10, beam_width=5):
   z = torch.randn(1, 1024)