Spaces:
Running
Running
junseok
commited on
Commit
·
92b8a7c
1
Parent(s):
b5cf8a0
final commit
Browse files- app.py +4 -4
- requirements.txt +2 -1
app.py
CHANGED
@@ -17,10 +17,10 @@ def calc_voxsim(inp_path, ref_path):
|
|
17 |
ref_wav = torch.FloatTensor(ref_wav)
|
18 |
|
19 |
with torch.no_grad():
|
20 |
-
input_emb_1 = F.normalize(model.
|
21 |
-
input_emb_2 = F.normalize(model.
|
22 |
-
ref_emb_1 = F.normalize(model.
|
23 |
-
ref_emb_2 = F.normalize(model.
|
24 |
|
25 |
score_1 = torch.mean(torch.matmul(input_emb_1, ref_emb_1.T))
|
26 |
score_2 = torch.mean(torch.matmul(input_emb_2, ref_emb_2.T))
|
|
|
17 |
ref_wav = torch.FloatTensor(ref_wav)
|
18 |
|
19 |
with torch.no_grad():
|
20 |
+
input_emb_1 = F.normalize(model.forward(inp_wavs), p=2, dim=1)
|
21 |
+
input_emb_2 = F.normalize(model.forward(inp_wav), p=2, dim=1)
|
22 |
+
ref_emb_1 = F.normalize(model.forward(ref_wavs), p=2, dim=1)
|
23 |
+
ref_emb_2 = F.normalize(model.forward(ref_wav), p=2, dim=1)
|
24 |
|
25 |
score_1 = torch.mean(torch.matmul(input_emb_1, ref_emb_1.T))
|
26 |
score_2 = torch.mean(torch.matmul(input_emb_2, ref_emb_2.T))
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ torch
|
|
4 |
torchaudio
|
5 |
tqdm
|
6 |
s3prl
|
7 |
-
huggingface_hub
|
|
|
|
4 |
torchaudio
|
5 |
tqdm
|
6 |
s3prl
|
7 |
+
huggingface_hub
|
8 |
+
gradio
|