Update app.py

app.py

@@ -17,15 +17,25 @@ def Read_DataSet(link):
         return df
 
 
+
 def remove_nn(wav, sample_rate=16000):
-    sf.write("full_generation.wav", wav, sample_rate, format='WAV')
-    audio, _ = load_audio('full_generation.wav', sr=df_state.sr())
+    
+    audio=librosa.resample(wav,orig_sr=sample_rate,target_sr=df_state.sr(),)
+    
+    audio=torch.tensor([audio])
+  #  audio, _ = load_audio('full_generation.wav', sr=df_state.sr())
+    print(audio)
+    
     enhanced = enhance(model_enhance, df_state, audio)
-    save_audio("enhanced.wav", enhanced, df_state.sr())
-    audiodata, samplerate = librosa.load("enhanced.wav", sr=16000)
+    print(enhanced)
+   # save_audio("enhanced.wav", enhanced, df_state.sr())
+    audiodata=librosa.resample(enhanced.numpy(),orig_sr=df_state.sr(),target_sr=sample_rate)
+   
     return 16000, audiodata
 
 
+
+
 class DataViewerApp:
     def __init__(self,df):
         #df=Read_DataSet(link)
@@ -80,8 +90,9 @@ class DataViewerApp:
         self.data.iloc[self.current_selected] = row
         sr,audio=data_oudio
         if sr!=16000:
-            sf.write("temp.wav", audio, sr,format='WAV')
-            audio, samplerate = librosa.load("temp.wav", sr=16000)
+            
+            audio=librosa.resample(audio,orig_sr=sr,target_sr=16000)
+   
 
         self.sdata[self.current_selected] = audio
         return self.get_page_data(self.current_page)