asigalov61 commited on
Commit
e469266
1 Parent(s): 71f8180

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -186,11 +186,18 @@ if __name__ == "__main__":
186
  parser.add_argument("--max-gen", type=int, default=1024, help="max")
187
  opt = parser.parse_args()
188
  soundfont_path = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
189
  models_info = {"generic pretrain model": ["skytnt/midi-model", ""],
190
  "j-pop finetune model": ["skytnt/midi-model-ft", "jpop/"],
191
  "touhou finetune model": ["skytnt/midi-model-ft", "touhou/"]}
192
 
193
 
 
 
 
 
 
 
194
  load_javascript()
195
  app = gr.Blocks()
196
  with app:
 
186
  parser.add_argument("--max-gen", type=int, default=1024, help="max")
187
  opt = parser.parse_args()
188
  soundfont_path = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
189
+ meta_data_path = "/meta-data/LAMD_META_10000.pickle"
190
  models_info = {"generic pretrain model": ["skytnt/midi-model", ""],
191
  "j-pop finetune model": ["skytnt/midi-model-ft", "jpop/"],
192
  "touhou finetune model": ["skytnt/midi-model-ft", "touhou/"]}
193
 
194
 
195
+ print('Loading meta-data...')
196
+ with open(meta_data_path, 'rb') as f:
197
+ meta_data = pickle.load(f)
198
+ print('Done!')
199
+
200
+
201
  load_javascript()
202
  app = gr.Blocks()
203
  with app: