Update app.py

app.py

@@ -164,11 +164,13 @@ def generate_music(
 
     mmtokenizer = _MMSentencePieceTokenizer("./mm_tokenizer_v0.2_hf/tokenizer.model")
     
-    basic_model_config='./xcodec_mini_infer/final_ckpt/config.yaml',
-    resume_path='./xcodec_mini_infer/final_ckpt/ckpt_00360000.pth',
-    config_path='./xcodec_mini_infer/decoders/config.yaml',
-    vocal_decoder_path='./xcodec_mini_infer/decoders/decoder_131000.pth',
-    inst_decoder_path='./xcodec_mini_infer/decoders/decoder_151000.pth',
+    codectool = CodecManipulator("xcodec", 0, 1)
+    model_config = OmegaConf.load(basic_model_config)
+    codec_model = eval(model_config.generator.name)(**model_config.generator.config).to(device)
+    parameter_dict = torch.load(resume_path, map_location='cpu')
+    codec_model.load_state_dict(parameter_dict['codec_model'])
+    codec_model.to(device)
+    codec_model.eval()
     
     class BlockTokenRangeProcessor(LogitsProcessor):
         def __init__(self, start_id, end_id):