Spaces:
Sleeping
Sleeping
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
|