Update app.py

app.py

@@ -143,8 +143,8 @@ re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_compl
 session, onnx_model, input_names, output_names = load_model(model_ver)
 
 if st.button('Сгенерировать потери'):
-    start_time = time.time()
     with st.spinner('Ожидайте...'):
+        start_time = time.time()
         output = inference(re_im, session, onnx_model, input_names, output_names)
         st.text(str(time.time() - start_time))
         st.subheader('3. Визуализация')