anton-l HF staff commited on
Commit
c4a1dc6
·
1 Parent(s): c4a4b81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
  import gradio as gr
3
  import torch
4
- from torchaudio.sox_effects import apply_effects_file
 
5
  from transformers import AutoFeatureExtractor, AutoModelForAudioXVector
6
 
7
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -55,8 +56,8 @@ def similarity_fn(path1, path2):
55
  if not (path1 and path2):
56
  return '<b style="color:red">ERROR: Please record audio for *both* speakers!</b>'
57
 
58
- wav1, _ = apply_effects_file(os.path.abspath(path1), EFFECTS, format="wav")
59
- wav2, _ = apply_effects_file(os.path.abspath(path2), EFFECTS, format="wav")
60
  print(wav1.shape, wav2.shape)
61
 
62
  input1 = feature_extractor(wav1.squeeze(0), return_tensors="pt", sampling_rate=16000).input_values.to(device)
 
1
  import os
2
  import gradio as gr
3
  import torch
4
+ import torchaudio
5
+ from torchaudio.sox_effects import apply_effects_tensor
6
  from transformers import AutoFeatureExtractor, AutoModelForAudioXVector
7
 
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
56
  if not (path1 and path2):
57
  return '<b style="color:red">ERROR: Please record audio for *both* speakers!</b>'
58
 
59
+ wav1, _ = apply_effects_file(*torchaudio.load(path1), EFFECTS)
60
+ wav2, _ = apply_effects_file(*torchaudio.load(path2), EFFECTS)
61
  print(wav1.shape, wav2.shape)
62
 
63
  input1 = feature_extractor(wav1.squeeze(0), return_tensors="pt", sampling_rate=16000).input_values.to(device)