Update app.py

app.py

@@ -26,7 +26,7 @@ from speechmos import plcmos
 
 import speech_recognition as speech_r
 from jiwer import wer
-
+import time
 
 @st.cache
 def load_model(model):
@@ -140,10 +140,10 @@ lossy_input_tensor = torch.tensor(lossy_input)
 re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_complex=False).permute(1, 0, 2).unsqueeze(
     1).numpy().astype(np.float32)
 
-
 session, onnx_model, input_names, output_names = load_model(model_ver)
 
 if st.button('Сгенерировать потери'):
+    start_time = time.time()
     with st.spinner('Ожидайте...'):
         output = inference(re_im, session, onnx_model, input_names, output_names)
 
@@ -374,5 +374,5 @@ if st.button('Сгенерировать потери'):
     df_1['WER'] = WER_mass
     
     st.dataframe(df_1)
-
+    print("--- %s seconds ---" % (time.time() - start_time))