Update sonogram_utility.py

sonogram_utility.py

@@ -129,8 +129,8 @@ def audioNormalize(waveform,sampleRate,stepSizeInSeconds = 2,dbThreshold = -50,d
     currEnd = int(min(currStart + stepSizeInSeconds * sampleRate, len(copyWaveform[0])-1))
     done = False
     while(not done):
-        print(f"{currStart//len(copyWaveform[0]):.2f}% equalized")
         copyWaveform_db = waveform[:,currStart:currEnd].clone().detach()
+        print(copyWaveform_db.shape)
         copyWaveform_db = transform(copyWaveform_db)
         if currStart == 0:
             print("First DB level calculated")
@@ -149,7 +149,7 @@ def audioNormalize(waveform,sampleRate,stepSizeInSeconds = 2,dbThreshold = -50,d
         if currStart > currEnd:
             done = True
         else:
-            currEnd = int(min(currStart + stepSizeInSeconds * sampleRate, len(copyWaveform_db[0])-1))
+            currEnd = int(min(currStart + stepSizeInSeconds * sampleRate, len(copyWaveform[0])-1))
     print("Waveform enhanced")
     return copyWaveform