Spaces:
Runtime error
Runtime error
update due to torchaudio version changes
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
-
|
|
|
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)
|