Update steganography.py

steganography.py

@@ -56,6 +56,10 @@ def create_text_image(text, font, base_width=512, height=256, margin=10, letter_
 def spectrogram_image_to_audio(image, sr=DEFAULT_SAMPLE_RATE):
     flipped_image = np.flipud(image)
     S = flipped_image.astype(np.float32) / 255.0 * 100.0
+
+    # Ensure the spectrogram has the correct shape
+    S = librosa.util.fix_length(S, size=N_MELS)
+
     y = librosa.griffinlim(S, n_iter=60, window='hann', n_fft=N_FFT, hop_length=N_FFT//4)
     return y