Champion commited on
Commit
87db0b4
·
verified ·
1 Parent(s): 5f41061

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ def inference(audio, model_tag="hifigan_bn_tdnnf_wav2vec2_vq_48_v1"):
82
  audio = torchaudio.transforms.Resample(orig_freq=sr,
83
  new_freq=16000)(audio)
84
  print(model_tag, file=sys.stderr)
85
- model = torch.hub.load("deep-privacy/SA-toolkit", "anonymization", tag_version=model_tag, trust_repo=True)
86
  model.eval()
87
  wav_conv = model.convert(audio, target="6081") # hard coded target
88
  return 16000, float2pcm(wav_conv.squeeze().cpu().numpy())
 
82
  audio = torchaudio.transforms.Resample(orig_freq=sr,
83
  new_freq=16000)(audio)
84
  print(model_tag, file=sys.stderr)
85
+ model = torch.hub.load("deep-privacy/SA-toolkit", "anonymization", tag_version=model_tag, trust_repo=True, force_reload=True)
86
  model.eval()
87
  wav_conv = model.convert(audio, target="6081") # hard coded target
88
  return 16000, float2pcm(wav_conv.squeeze().cpu().numpy())