Update README.md

README.md

@@ -62,13 +62,13 @@ for trans in transforms:
     lms = trans(lms)
 
 # 2. template of input
-input = dict()
-input['filenames'] = [music_path.split('/')[-1]]
-input['ans_crds'] = [0]
-input['audio_crds'] = [0]
-input['attention_mask'] = torch.tensor([[1, 1, 1, 1, 1]]).to(device)
-input['input_ids'] = torch.tensor([[1, 694, 5777, 683, 13]]).to(device)
-input['spectrogram'] = torch.from_numpy(lms).unsqueez(dim=0).to(device)
+input_dic = dict()
+input_dic['filenames'] = [music_path.split('/')[-1]]
+input_dic['ans_crds'] = [0]
+input_dic['audio_crds'] = [0]
+input_dic['attention_mask'] = torch.tensor([[1, 1, 1, 1, 1]]).to(device)
+input_dic['input_ids'] = torch.tensor([[1, 694, 5777, 683, 13]]).to(device)
+input_dic['spectrogram'] = torch.from_numpy(lms).unsqueez(dim=0).to(device)
 # 3. generation
 model.eval()
 gen_ids = model.forward_test(input)