Update goai_helpers/utils.py

goai_helpers/utils.py

@@ -14,8 +14,6 @@ from TTS.tts.layers.xtts.tokenizer import VoiceBpeTokenizer
 from TTS.tts.configs.xtts_config import XttsConfig
 from TTS.tts.models.xtts import Xtts
 
-from resemble_enhance.enhancer.inference import denoise, enhance
-
 
 def download_file(url: str, destination: str, token: str = None):
     """
@@ -74,37 +72,3 @@ def diviser_phrases_moore(texte: str) -> list:
     
     return phrases
 
-
-# function to enhance speech
-@spaces.GPU
-def enhance_speech(audio_array, sampling_rate, solver, nfe, tau, denoise_before_enhancement):
-    solver = solver.lower()
-    nfe = int(nfe)
-    lambd = 0.9 if denoise_before_enhancement else 0.1
-
-    # device
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-    def denoise_audio():
-        try:
-            return denoise(audio_array, sampling_rate, device)
-        except Exception as e:
-            print("> Error while denoising : ", str(e))
-            return audio_array, sampling_rate
-
-    def enhance_audio():
-        try:
-            return enhance(audio_array, sampling_rate, device, nfe=nfe, solver=solver, lambd=lambd, tau=tau)
-        except Exception as e:
-            print("> Error while enhancement : ", str(e))
-            return audio_array, sampling_rate
-
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        future_denoise = executor.submit(denoise_audio)
-        future_enhance = executor.submit(enhance_audio)
-
-        denoised_audio, new_sr1 = future_denoise.result()
-        enhanced_audio, new_sr2 = future_enhance.result()
-
-        # convert to numpy and return
-        return (new_sr1, denoised_audio.cpu().numpy()), (new_sr2, enhanced_audio.cpu().numpy())