modify app v2

app.py

@@ -121,7 +121,7 @@ def gen_semantic_feature(frame_dir, semantic_dir):
             features[idx] = image_features[0]
         features = features.numpy()
         np.save(output_path, features)
-        
+
 def gen_emotion_feature(frame_dir, emotion_dir):
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     model, preprocess = clip.load("ViT-L/14@336px", device=device)
@@ -460,14 +460,15 @@ class Video2music:
             chordRootDic = json.load(json_file)
         with open('dataset/vevo_meta/chord_attr.json') as json_file:
             chordAttrDic = json.load(json_file)
-        
+
+        primer = primer.strip()
         if primer.strip() == "":
             if "major" in key:
                 primer = "C"
             else:
                 primer = "Am"
         
-        pChordList = primer.split(" ")
+        pChordList = primer.split()
 
         primerCID = []
         primerCID_root = []
@@ -497,9 +498,10 @@ class Video2music:
                 if pChord[type_idx+1:] == "":
                     pChord = pChord[0:type_idx]
             
+            print("pchord is ", pChord)
             chordID = chordDic[pChord]
             primerCID.append(chordID)
-
+            
             chord_arr = pChord.split(":")
             if len(chord_arr) == 1:
                 chordRootID = chordRootDic[chord_arr[0]]