Update extract_feature_print.py

extract_feature_print.py

@@ -190,7 +190,10 @@ os.makedirs(outPath, exist_ok=True)
 def readwave(wav_path, normalize=False):
     wav, sr = sf.read(wav_path)
     assert sr == 16000
-    feats = torch.from_numpy(wav).float()
+    if Custom_Embed:
+        feats = torch.from_numpy(wav).float()
+    else:
+        feats = torch.from_numpy(load_audio(wav, sr)).to(dtype).to(device)
     if feats.dim() == 2:  # double channels
         feats = feats.mean(-1)
     assert feats.dim() == 1, feats.dim()