Update app.py

app.py

@@ -43,8 +43,8 @@ def inference(re_im, session, onnx_model, input_names, output_names):
     output_audio = torch.istft(output_audio, window, stride, window=hann)
     return output_audio.numpy()
 
-def visualize(hr, lr, recon):
-    sr = CONFIG.DATA.sr
+def visualize(hr, lr, recon, sr):
+    sr = sr
     window_size = 1024
     window = np.hanning(window_size)
 
@@ -108,7 +108,7 @@ if st.button('Сгенерировать потери'):
         output = inference(re_im, session, onnx_model, input_names, output_names)
 
         st.subheader('3. Визуализация')
-        fig = visualize(target, lossy_input, output)
+        fig = visualize(target, lossy_input, output, sr)
         st.pyplot(fig)
     st.success('Сделано!')
     sf.write('target.wav', target, sr)