XDHDD commited on
Commit
70e239c
1 Parent(s): 0a725ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -143,8 +143,8 @@ re_im = torch.stft(lossy_input_tensor, window, stride, window=hann, return_compl
143
  session, onnx_model, input_names, output_names = load_model(model_ver)
144
 
145
  if st.button('Сгенерировать потери'):
146
- start_time = time.time()
147
  with st.spinner('Ожидайте...'):
 
148
  output = inference(re_im, session, onnx_model, input_names, output_names)
149
  st.text(str(time.time() - start_time))
150
  st.subheader('3. Визуализация')
 
143
  session, onnx_model, input_names, output_names = load_model(model_ver)
144
 
145
  if st.button('Сгенерировать потери'):
 
146
  with st.spinner('Ожидайте...'):
147
+ start_time = time.time()
148
  output = inference(re_im, session, onnx_model, input_names, output_names)
149
  st.text(str(time.time() - start_time))
150
  st.subheader('3. Визуализация')