update

app.py

@@ -2,11 +2,13 @@ from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
 import gradio as gr
 import numpy as np
-import warnings 
+import warnings
+from google.cloud import storage
 
 
 class MusicGenHandler():
-    def __init__(self, init_model_path='./cs-pretrained/stem_model', generation_duration=30.0):
+    # def __init__(self, init_model_path='./cs-pretrained/stem_model', generation_duration=30.0):
+    def __init__(self, init_model_path='createsafe/grimes-stem-model', generation_duration=30.0):
         self.model_path = init_model_path
         self.generation_duration = generation_duration
         self._setup_model()