Update app.py

app.py

@@ -20,6 +20,7 @@ def Read_DataSet(link):
 
 
 
+
 def remove_nn(wav, sample_rate=16000):
     
     audio=librosa.resample(wav,orig_sr=sample_rate,target_sr=df_state.sr(),)
@@ -31,9 +32,9 @@ def remove_nn(wav, sample_rate=16000):
     enhanced = enhance(model_enhance, df_state, audio)
     print(enhanced)
    # save_audio("enhanced.wav", enhanced, df_state.sr())
-    audiodata=librosa.resample(enhanced.numpy(),orig_sr=df_state.sr(),target_sr=sample_rate)
+    audiodata=librosa.resample(enhanced[0].numpy(),orig_sr=df_state.sr(),target_sr=sample_rate)
    
-    return 16000, audiodata
+    return 16000, audiodata/np.max(audiodata)
 
 
 
@@ -94,6 +95,8 @@ class DataViewerApp:
         if sr!=16000:
             
             audio=librosa.resample(audio,orig_sr=sr,target_sr=16000)
+            audio=np.float32(audio)
+            audio/=np.max(audio)
    
 
         self.sdata[self.current_selected] = audio