junseok commited on
Commit
92b8a7c
·
1 Parent(s): b5cf8a0

final commit

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. 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.foward(inp_wavs), p=2, dim=1)
21
- input_emb_2 = F.normalize(model.foward(inp_wav), p=2, dim=1)
22
- ref_emb_1 = F.normalize(model.foward(ref_wavs), p=2, dim=1)
23
- ref_emb_2 = F.normalize(model.foward(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))
 
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