Update app.py

app.py

@@ -11,7 +11,12 @@ from typing import Any, Callable, Optional, Tuple, Union,Iterator
 import numpy as np
 import torch.nn as nn # Import the missing module
 
+import noisereduce as nr
 
+def remove_noise_nr(audio_data,sr=16000):
+    """يزيل الضوضاء باستخدام مكتبة noisereduce."""
+    reduced_noise = nr.reduce_noise(y=audio_data, sr=sr)
+    return reduced_noise
 
 def _inference_forward_stream(
         self,
@@ -139,7 +144,7 @@ def generate_audio_ai(text,name_model):
     #torch.cuda.empty_cache()
     with torch.no_grad():
         for chunk in _inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= speaker_embeddings,chunk_size=256):
-            yield  16000,chunk#.cpu().numpy().squeeze()#.astype(np.int16).tobytes()    
+            yield  16000,remove_noise_nr(chunk)#.cpu().numpy().squeeze()#.astype(np.int16).tobytes()    
 
     # yield generate_audio(text_answer,name_model)
 def remove_extra_spaces(text):