fix compat for app outside of HF
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ from audiocraft.models import MusicGen
|
|
15 |
from audiocraft.data.audio import audio_write
|
16 |
|
17 |
MODEL = None
|
18 |
-
IS_SHARED_SPACE = "musicgen/MusicGen" in os.environ
|
19 |
|
20 |
|
21 |
def load_model(version):
|
|
|
15 |
from audiocraft.data.audio import audio_write
|
16 |
|
17 |
MODEL = None
|
18 |
+
IS_SHARED_SPACE = "musicgen/MusicGen" in os.environ.get('SPACE_ID')
|
19 |
|
20 |
|
21 |
def load_model(version):
|