Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import pandas as pd
 from sentence_transformers import SentenceTransformer, util
 
 model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
@@ -17,5 +18,8 @@ if st.button("Guess"):
     word_embedding = model.encode(word)
     similarity = util.pytorch_cos_sim(secred_embedding, word_embedding).cpu().numpy()[0][0]
     st.session_state['words'].append((word, similarity))
-    words = st.session_state['words']
-    st.write(words)
+    words_df = pd.DataFrame(
+        st.session_state['words'],
+        columns=["word", "similarity"]
+    ).sort_values(by=["similarity"], ascending=False)
+    st.dataframe(words_df)