Updated line 112 with: ax.annotate("+ Inf", (trellis.size(0) - trellis.size(1) / 2.4 , trellis.size(1) / 3)) | Updated lines 260-261 with: updated_clean_UPPERCASE_transcript | Added Abby_and_Prince.jpg

app.py

@@ -108,8 +108,8 @@ def n_inf_to_p_inf():
     fig, ax = plt.subplots()
     img = ax.imshow(trellis.T, origin="lower")
     ax.annotate("- Inf", (trellis.size(1) / 5, trellis.size(1) / 1.5))
-    # Shift the "+ Inf" annotation to the left by decreasing the x-coordinate value
-    ax.annotate("+ Inf", (trellis.size(0) - trellis.size(1) / 1.4, trellis.size(1) / 3))
+    # Shift the "+ Inf" annotation to the right by increasing the denominator
+    ax.annotate("+ Inf", (trellis.size(0) - trellis.size(1) / 2.4 , trellis.size(1) / 3))
     fig.colorbar(img, ax=ax, shrink=0.25, location="bottom")
     fig.tight_layout()
     return fig
@@ -200,7 +200,7 @@ def merge_repeats(path):
         score = sum(path[k].score for k in range(i1, i2)) / (i2 - i1)
         segments.append(
             Segment(
-                transcript[path[i1].token_index],
+                updated_clean_UPPERCASE_transcript[path[i1].token_index],
                 path[i1].time_index,
                 path[i2 - 1].time_index + 1,
                 score,
@@ -257,8 +257,8 @@ def plot_trellis_with_segments(trellis, segments, transcript):
     return fig
 
 
-plot_trellis_with_segments(trellis, segments, clean_transcript)
-st.pyplot(plot_trellis_with_segments(trellis, segments, clean_transcript))
+plot_trellis_with_segments(trellis, segments, updated_clean_UPPERCASE_transcript)
+st.pyplot(plot_trellis_with_segments(trellis, segments, updated_clean_UPPERCASE_transcript))
 
 # Part L: Merge words | Segments
 # Merge words