Update demos/musicgen_app.py

demos/musicgen_app.py

@@ -22,11 +22,13 @@ from tempfile import NamedTemporaryFile
 import gradio as gr
 import torch
 from einops import rearrange
+from omegaconf import DictConfig, OmegaConf
 
 from audiocraft.data.audio import audio_write
 from audiocraft.data.audio_utils import convert_audio
 from audiocraft.models import MultiBandDiffusion, MusicGen
-from audiocraft.models.encodec import InterleaveStereoCompressionModel
+from audiocraft.models.builders import get_lm_model
+from audiocraft.models.encodec import CompressionModel, InterleaveStereoCompressionModel
 
 MODEL = None  # Last used model
 SPACE_ID = os.environ.get("SPACE_ID", "")