Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 
 # Import the model
-from jukebox.models import MODELS
+from models import MODELS
 
 model = jukebox.make_vqvae(MODELS['5B_LYRICS'], device="cpu")