hieugiaosu commited on
Commit
e47da3a
·
1 Parent(s): 07d805d
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -40,6 +40,8 @@ def seprate(mixed_voice_path, clean_voice_path, drop_down):
40
  mixed_voice = mixed_voice_tool
41
  else:
42
  mixed_voice,rate = load_voice(mixed_voice_path)
 
 
43
  clean_voice,rate = load_voice(clean_voice_path)
44
  if clean_voice.shape[-1] < 16000*4:
45
  n = 16000*4 // clean_voice.shape[-1] + 1
 
40
  mixed_voice = mixed_voice_tool
41
  else:
42
  mixed_voice,rate = load_voice(mixed_voice_path)
43
+ mixed_voice = mixed_voice.float()
44
+ mixed_voice = torch.nn.functional.normalize(mixed_voice, p=2, dim=-1)
45
  clean_voice,rate = load_voice(clean_voice_path)
46
  if clean_voice.shape[-1] < 16000*4:
47
  n = 16000*4 // clean_voice.shape[-1] + 1