Update app.py

app.py

@@ -72,19 +72,20 @@ def visualize(hr, lr, recon, sr):
 
     fig, (ax1, ax2, ax3) = plt.subplots(3, 1, sharey=True, sharex=True, figsize=(16, 12))
     ax1.title.set_text('Оригинальный сигнал')
-    ax1.set_xlabel('Время, с')
-    ax1.set_ylabel('Частота, Гц')
     ax2.title.set_text('Сигнал с потерями')
-    ax2.set_xlabel('Время, с')
-    ax2.set_ylabel('Частота, Гц')
     ax3.title.set_text('Улучшенный сигнал')
-    ax3.set_xlabel('Время, с')
-    ax3.set_ylabel('Частота, Гц')
 
     canvas = FigureCanvas(fig)
     p = librosa.display.specshow(librosa.amplitude_to_db(stft_hr), ax=ax1, y_axis='log', x_axis='time', sr=sr)
     p = librosa.display.specshow(librosa.amplitude_to_db(stft_lr), ax=ax2, y_axis='log', x_axis='time', sr=sr)
     p = librosa.display.specshow(librosa.amplitude_to_db(stft_recon), ax=ax3, y_axis='log', x_axis='time', sr=sr)
+
+    ax1.set_xlabel('Время, с')
+    ax1.set_ylabel('Частота, Гц')
+    ax2.set_xlabel('Время, с')
+    ax2.set_ylabel('Частота, Гц')
+    ax3.set_xlabel('Время, с')
+    ax3.set_ylabel('Частота, Гц')
     return fig
 
 packet_size = CONFIG.DATA.EVAL.packet_size