Aboubacar OUATTARA - kaira commited on
Commit
34f6b5d
1 Parent(s): e5f089f

add audios files

Browse files
Files changed (1) hide show
  1. app.py +37 -25
app.py CHANGED
@@ -63,37 +63,49 @@ def text_to_speech(bambara_text, reference_audio: Optional[Tuple] = None):
63
 
64
 
65
  # Function to enhance speech
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
66
  @spaces.GPU
67
  def enhance_speech(audio_array, sampling_rate, solver, nfe, tau, denoise_before_enhancement):
68
  solver = solver.lower()
69
  nfe = int(nfe)
70
  lambd = 0.9 if denoise_before_enhancement else 0.1
71
 
72
- @spaces.GPU(duration=360)
73
- def denoise_audio():
74
- try:
75
- return denoise(audio_array, sampling_rate, device)
76
- except Exception as e:
77
- print("> Error while denoising : ", str(e))
78
- return audio_array, sampling_rate
79
-
80
- @spaces.GPU(duration=360)
81
- def enhance_audio():
82
- try:
83
- return enhance(audio_array, sampling_rate, device, nfe=nfe, solver=solver, lambd=lambd, tau=tau)
84
- except Exception as e:
85
- print("> Error while enhancement : ", str(e))
86
- return audio_array, sampling_rate
87
-
88
- with concurrent.futures.ThreadPoolExecutor() as executor:
89
- future_denoise = executor.submit(denoise_audio)
90
- future_enhance = executor.submit(enhance_audio)
91
-
92
- denoised_audio, new_sr1 = future_denoise.result()
93
- enhanced_audio, new_sr2 = future_enhance.result()
94
-
95
- # Convert to numpy and return
96
- return (new_sr1, denoised_audio.cpu().numpy()), (new_sr2, enhanced_audio.cpu().numpy())
97
 
98
 
99
  def convert_to_int16(audio_array):
 
63
 
64
 
65
  # Function to enhance speech
66
+ # @spaces.GPU
67
+ # def enhance_speech(audio_array, sampling_rate, solver, nfe, tau, denoise_before_enhancement):
68
+ # solver = solver.lower()
69
+ # nfe = int(nfe)
70
+ # lambd = 0.9 if denoise_before_enhancement else 0.1
71
+ #
72
+ # @spaces.GPU(duration=360)
73
+ # def denoise_audio():
74
+ # try:
75
+ # return denoise(audio_array, sampling_rate, device)
76
+ # except Exception as e:
77
+ # print("> Error while denoising : ", str(e))
78
+ # return audio_array, sampling_rate
79
+ #
80
+ # @spaces.GPU(duration=360)
81
+ # def enhance_audio():
82
+ # try:
83
+ # return enhance(audio_array, sampling_rate, device, nfe=nfe, solver=solver, lambd=lambd, tau=tau)
84
+ # except Exception as e:
85
+ # print("> Error while enhancement : ", str(e))
86
+ # return audio_array, sampling_rate
87
+ #
88
+ # with concurrent.futures.ThreadPoolExecutor() as executor:
89
+ # future_denoise = executor.submit(denoise_audio)
90
+ # future_enhance = executor.submit(enhance_audio)
91
+ #
92
+ # denoised_audio, new_sr1 = future_denoise.result()
93
+ # enhanced_audio, new_sr2 = future_enhance.result()
94
+ #
95
+ # # Convert to numpy and return
96
+ # return (new_sr1, denoised_audio.cpu().numpy()), (new_sr2, enhanced_audio.cpu().numpy())
97
+
98
  @spaces.GPU
99
  def enhance_speech(audio_array, sampling_rate, solver, nfe, tau, denoise_before_enhancement):
100
  solver = solver.lower()
101
  nfe = int(nfe)
102
  lambd = 0.9 if denoise_before_enhancement else 0.1
103
 
104
+ denoised_audio, new_sr1 = denoise(audio_array, sampling_rate, device)
105
+ enhanced_audio, new_sr2 = enhance(audio_array, sampling_rate, device, nfe=nfe, solver=solver, lambd=lambd, tau=tau)
106
+
107
+ # Convert to numpy and return
108
+ return (new_sr1, denoised_audio.cpu().numpy()), (new_sr2, enhanced_audio.cpu().numpy())
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
109
 
110
 
111
  def convert_to_int16(audio_array):