Spaces:
Running
on
Zero
Running
on
Zero
Aboubacar OUATTARA - kaira
commited on
Commit
•
34f6b5d
1
Parent(s):
e5f089f
add audios files
Browse files
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 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
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):
|