Update utils.py

underworld02 · Aug 23, 2022 · b31b7ea · b31b7ea
1 parent a4a33fb
commit b31b7ea
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -1,7 +1,7 @@
 import logging
 from json import loads
 from torch import load
-import logging
+import l
 
 
 class HParams():
@@ -67,3 +67,8 @@ def get_hparams_from_file(config_path):
 
   hparams = HParams(**config)
   return hparams
+
+
+def load_audio_to_torch(full_path, target_sampling_rate):
+  audio, sampling_rate = librosa.load(full_path, sr=target_sampling_rate, mono=True)
+  return torch.FloatTensor(audio.astype(np.float32))