XDHDD commited on
Commit
6fe43d7
1 Parent(s): 1d1e6d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -57,7 +57,7 @@ def visualize(hr, lr, recon):
57
  stft_recon = librosa.core.spectrum.stft(recon, n_fft=window_size, hop_length=512, window=window)
58
  stft_recon = 2 * np.abs(stft_recon) / np.sum(window)
59
 
60
- fig, (ax1, ax2, ax3) = plt.subplots(3, 1, sharey=True, sharex=True, figsize=(16, 10))
61
  ax1.title.set_text('Оригинальный сигнал')
62
  ax2.title.set_text('Сигнал с потерями')
63
  ax3.title.set_text('Улучшенный сигнал')
@@ -76,7 +76,7 @@ title = 'Сокрытие потерь пакетов'
76
  st.set_page_config(page_title=title, page_icon=":sound:")
77
  st.title(title)
78
 
79
- st.subheader('Загрузка аудио')
80
  uploaded_file = st.file_uploader("Загрузите аудио формата (.wav) 48 КГц")
81
 
82
  is_file_uploaded = uploaded_file is not None
@@ -89,7 +89,7 @@ target = target[:packet_size * (len(target) // packet_size)]
89
  st.text('Ваше аудио')
90
  st.audio(uploaded_file)
91
 
92
- st.subheader('Выберите желаемый процент потерь')
93
  slider = [st.slider("Ожидаемый процент потерь для генератора потерь цепи Маркова", 0, 100, step=1)]
94
  loss_percent = float(slider[0])/100
95
  mask_gen = MaskGenerator(is_train=False, probs=[(1 - loss_percent, loss_percent)])
@@ -107,7 +107,7 @@ if st.button('Сгенерировать потери'):
107
  with st.spinner('Ожидайте...'):
108
  output = inference(re_im, session, onnx_model, input_names, output_names)
109
 
110
- st.subheader('Визуализация')
111
  fig = visualize(target, lossy_input, output)
112
  st.pyplot(fig)
113
  st.success('Сделано!')
 
57
  stft_recon = librosa.core.spectrum.stft(recon, n_fft=window_size, hop_length=512, window=window)
58
  stft_recon = 2 * np.abs(stft_recon) / np.sum(window)
59
 
60
+ fig, (ax1, ax2, ax3) = plt.subplots(3, 1, sharey=True, sharex=True, figsize=(16, 20))
61
  ax1.title.set_text('Оригинальный сигнал')
62
  ax2.title.set_text('Сигнал с потерями')
63
  ax3.title.set_text('Улучшенный сигнал')
 
76
  st.set_page_config(page_title=title, page_icon=":sound:")
77
  st.title(title)
78
 
79
+ st.subheader('1. Загрузка аудио')
80
  uploaded_file = st.file_uploader("Загрузите аудио формата (.wav) 48 КГц")
81
 
82
  is_file_uploaded = uploaded_file is not None
 
89
  st.text('Ваше аудио')
90
  st.audio(uploaded_file)
91
 
92
+ st.subheader('2. Выберите желаемый процент потерь')
93
  slider = [st.slider("Ожидаемый процент потерь для генератора потерь цепи Маркова", 0, 100, step=1)]
94
  loss_percent = float(slider[0])/100
95
  mask_gen = MaskGenerator(is_train=False, probs=[(1 - loss_percent, loss_percent)])
 
107
  with st.spinner('Ожидайте...'):
108
  output = inference(re_im, session, onnx_model, input_names, output_names)
109
 
110
+ st.subheader('3. Визуализация')
111
  fig = visualize(target, lossy_input, output)
112
  st.pyplot(fig)
113
  st.success('Сделано!')