chong.zhang commited on
Commit
f2759a6
·
1 Parent(s): 7724234
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +5 -4
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
  title: InspireMusic
3
  emoji: 🎶
4
- colorFrom: blue
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 model_name == "InspireMusic-Base":
195
- snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
196
- else:
197
- snapshot_download(f"iic/{model_name}", local_dir=model_dir)
 
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()