How do you determing the correct dimension of Mel Spectrogram Feature Extraction for NN

Question

I trying to implement a Mel Spectrogram feature extraction:

n_mels = 128

# Extracting MelFrequency Spectrum for every file
def extract_features(file_name):
t