Update app.py

app.py

@@ -5,6 +5,9 @@ import torchaudio
 import os
 import numpy as np
 import base64
+from audiocraft.models import MAGNeT
+from audiocraft.data.audio import audio_write
+
 from audiocraft.data.audio_utils import convert_audio
 from audiocraft.data.audio import audio_write
 from audiocraft.models.encodec import InterleaveStereoCompressionModel
@@ -14,7 +17,8 @@ genres = ["Pop","Hip-Hop", "Classical","Lofi", "Chillpop","Country","R&G", "Folk
 
 @st.cache_resource()
 def load_model():
-    model = MusicGen.get_pretrained('facebook/audiogen-medium')
+    model = MAGNeT.get_pretrained("facebook/magnet-medium-30secs")
+    # model = MusicGen.get_pretrained('facebook/audiogen-medium')
     return model