Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
f2759a6
1
Parent(s):
7724234
update
Browse files
README.md
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
---
|
2 |
title: InspireMusic
|
3 |
emoji: 🎶
|
4 |
-
colorFrom:
|
5 |
colorTo: purple
|
6 |
sdk: gradio
|
7 |
sdk_version: 5.8.0
|
|
|
1 |
---
|
2 |
title: InspireMusic
|
3 |
emoji: 🎶
|
4 |
+
colorFrom: indigo
|
5 |
colorTo: purple
|
6 |
sdk: gradio
|
7 |
sdk_version: 5.8.0
|
app.py
CHANGED
@@ -191,8 +191,9 @@ if __name__ == '__main__':
|
|
191 |
|
192 |
for model_name in model_list:
|
193 |
model_dir = f"pretrained_models/{model_name}"
|
194 |
-
if
|
195 |
-
|
196 |
-
|
197 |
-
|
|
|
198 |
main()
|
|
|
191 |
|
192 |
for model_name in model_list:
|
193 |
model_dir = f"pretrained_models/{model_name}"
|
194 |
+
if not os.path.isdir(model_dir):
|
195 |
+
if model_name == "InspireMusic-Base":
|
196 |
+
snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
|
197 |
+
else:
|
198 |
+
snapshot_download(f"iic/{model_name}", local_dir=model_dir)
|
199 |
main()
|