Update app.py

app.py

@@ -14,6 +14,12 @@ import numpy as np
 from matplotlib import pyplot as plt
 from matplotlib import gridspec
 
+# Adjust vertical spacing between subplots
+plt.subplots_adjust(hspace=0.15)  # You can adjust the value as needed
+
+# Adjust the white space (margins) around the plot
+plt.tight_layout(pad=0.5)  # You can adjust the pad value as needed
+
 torch.manual_seed(1234)
 MAX_WAV_VALUE = 32768.0
 
@@ -55,8 +61,6 @@ def plot_spec_align(mel, align):
     ax_align.imshow(align)
     ax_align.set_title('Alignment', fontsize=12)
 
-    plt.subplots_adjust(hspace=0.15)
-
     return fig_mel # fig_align