4kasha commited on
Commit
8229cee
·
1 Parent(s): f31ab4f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -134,11 +134,11 @@ def main():
134
  align_matrix, cost_matrix, loss, similarity_matrix = aligner.compute_alignment_matrixes(s1_vec, s2_vec)
135
  print(align_matrix.shape, cost_matrix.shape)
136
 
137
- st.write(f"**word alignment matrix** (loss: :blue[{loss}])")
138
  fig = plot_align_matrix_heatmap_plotly(align_matrix.T, sent1, sent2, threshhold, cost_matrix.T)
139
  st.plotly_chart(fig, use_container_width=True)
140
 
141
- st.write(f"**word similarity matrix**")
142
  fig2 = plot_similarity_matrix_heatmap_plotly(similarity_matrix.T, sent1, sent2, cost_matrix.T)
143
  st.plotly_chart(fig2, use_container_width=True)
144
 
 
134
  align_matrix, cost_matrix, loss, similarity_matrix = aligner.compute_alignment_matrixes(s1_vec, s2_vec)
135
  print(align_matrix.shape, cost_matrix.shape)
136
 
137
+ st.write(f"**Word alignment matrix** (loss: :blue[{loss}])")
138
  fig = plot_align_matrix_heatmap_plotly(align_matrix.T, sent1, sent2, threshhold, cost_matrix.T)
139
  st.plotly_chart(fig, use_container_width=True)
140
 
141
+ st.write(f"**Word similarity matrix**")
142
  fig2 = plot_similarity_matrix_heatmap_plotly(similarity_matrix.T, sent1, sent2, cost_matrix.T)
143
  st.plotly_chart(fig2, use_container_width=True)
144