Update app.py

app.py

@@ -63,9 +63,9 @@ def visualize(hr, lr, recon):
     ax3.title.set_text('Улучшенный сигнал')
 
     canvas = FigureCanvas(fig)
-    p = librosa.display.specshow(librosa.amplitude_to_db(stft_hr, ref=np.max), ax=ax1, y_axis='log', x_axis='time', sr=sr)
-    p = librosa.display.specshow(librosa.amplitude_to_db(stft_lr, ref=np.max), ax=ax2, y_axis='log', x_axis='time', sr=sr)
-    p = librosa.display.specshow(librosa.amplitude_to_db(stft_recon, ref=np.max), ax=ax3, y_axis='log', x_axis='time', sr=sr)
+    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)
     return fig
 
 packet_size = CONFIG.DATA.EVAL.packet_size