hubert load_audio

Author: MaxMax2016

hubert/inference.py

@@ -3,11 +3,16 @@
 import numpy as np
 import argparse
 import torch
+import librosa
 
-from whisper.audio import load_audio
 from hubert import hubert_model
 
 
+def load_audio(file: str, sr: int = 16000):
+    x, sr = librosa.load(file, sr=sr)
+    return x
+
+
 def load_model(path, device):
     model = hubert_model.hubert_soft(path)
     model.eval()