Bagus commited on
Commit
09b7c54
·
1 Parent(s): 15edccd

update due to torchaudio version changes

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