Update app.py

app.py

@@ -146,12 +146,11 @@ if st.button('Сгенерировать потери'):
     start_time = time.time()
     with st.spinner('Ожидайте...'):
         output = inference(re_im, session, onnx_model, input_names, output_names)
-
+        st.text(str(time.time() - start_time))
         st.subheader('3. Визуализация')
         fig = visualize(target, lossy_input, output, sr)
         st.pyplot(fig)
     st.success('Сделано!')
-    st.text(str(time.time() - start_time))
     sf.write('target.wav', target, sr)
     sf.write('lossy.wav', lossy_input, sr)
     sf.write('enhanced.wav', output, sr)