4kasha
commited on
Commit
·
8229cee
1
Parent(s):
f31ab4f
minor fix
Browse files
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"**
|
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"**
|
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 |
|