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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,8 +55,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)
59
- wav2, _ = apply_effects_file(os.path.abspath(path2), EFFECTS)
60
  print(wav1.shape, wav2.shape)
61
 
62
  input1 = feature_extractor(wav1.squeeze(0), return_tensors="pt", sampling_rate=16000).input_values.to(device)
 
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)