Kevin676 commited on
Commit
26e0b3b
·
1 Parent(s): ff9ad54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -71,7 +71,7 @@ def transcribe(apikey, upload, audio, choice1):
71
  rate, data = wavfile.read("output.wav")
72
 
73
  #reduced_noise = nr.reduce_noise(y=data, sr=rate, prop_decrease= 0.9, stationary=True)
74
- reduced_noise = nr.reduce_noise(y = data, sr=rate, prop_decrease= 0.85, thresh_n_mult_nonstationary=2, stationary=False)
75
  #reduced_noise = nr.reduce_noise(y = data, sr=rate, thresh_n_mult_nonstationary=2, stationary=False)
76
 
77
  wavfile.write("audio1.wav", rate, reduced_noise)
@@ -87,8 +87,7 @@ def transcribe(apikey, upload, audio, choice1):
87
  ).unsqueeze(0)
88
 
89
  enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
90
- #torchaudio.save("enhanced.wav", enhanced.cpu(), 16000)
91
- torchaudio.save("enhanced.wav", enhanced.cpu(), 24000)
92
 
93
  return [result.text, chat_response, "enhanced.wav"]
94
 
 
71
  rate, data = wavfile.read("output.wav")
72
 
73
  #reduced_noise = nr.reduce_noise(y=data, sr=rate, prop_decrease= 0.9, stationary=True)
74
+ reduced_noise = nr.reduce_noise(y = data, sr=rate, prop_decrease= 0.9, thresh_n_mult_nonstationary=2, stationary=False)
75
  #reduced_noise = nr.reduce_noise(y = data, sr=rate, thresh_n_mult_nonstationary=2, stationary=False)
76
 
77
  wavfile.write("audio1.wav", rate, reduced_noise)
 
87
  ).unsqueeze(0)
88
 
89
  enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
90
+ torchaudio.save("enhanced.wav", enhanced.cpu(), 16000)
 
91
 
92
  return [result.text, chat_response, "enhanced.wav"]
93