Updated lines 179 and 335

app.py

@@ -176,7 +176,7 @@ def plot_trellis_with_path(trellis, path):
     fig.tight_layout()
     return fig
     
-st.pyplot(fig)
+st.pyplot(plot_trellis_with_path(trellis, path))
 
 # Part J: Merge Repeats | Segments
 # Merge the labels
@@ -261,7 +261,7 @@ def plot_alignments(trellis, segments, word_segments, waveform=np.random.randn(1
     fig.tight_layout()
     return fig
 
-st.pyplot(fig)
+st.pyplot()
 
 # Part L: Merge words | Segments
 # Merge words
@@ -331,8 +331,8 @@ def plot_alignments(trellis, segments, word_segments, waveform=np.random.randn(1
     return fig
 
 
-plot_alignments(trellis, segments, word_segments, waveform, sample_rate)
-st.pyplot(plot_alignments(trellis, word_segments, waveform, sample_rate))
+#plot_alignments(trellis, segments, word_segments, waveform, sample_rate)
+st.pyplot(plot_alignments(trellis, segments, word_segments, waveform=np.random.randn(1024), sample_rate=44100))
 
 # Part N: Display Segment
 def display_segment(i):