Update extract_feature_print.py

extract_feature_print.py

@@ -1,6 +1,6 @@
 import os, sys, traceback
 from transformers import HubertModel
-import os
+import librosa
 from torch import nn
 import torch
 
@@ -157,6 +157,19 @@ class Config:
         )
 config = Config()
 
+def load_audio(file, sample_rate):
+    try:
+        file = file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
+        audio, sr = sf.read(file)
+        if len(audio.shape) > 1:
+            audio = librosa.to_mono(audio.T)
+        if sr != sample_rate:
+            audio = librosa.resample(audio, orig_sr=sr, target_sr=sample_rate)
+    except Exception as error:
+        raise RuntimeError(f"An error occurred loading the audio: {error}")
+
+    return audio.flatten()
+
 #HuggingFacePlaceHolder = None
 class HubertModelWithFinalProj(HubertModel):
     def __init__(self, config):