Update app.py

app.py

@@ -25,8 +25,9 @@ if st.button("Guess") or word:
         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_df = pd.DataFrame(
-            st.session_state['words'],
-            columns=["word", "similarity"]
-        ).sort_values(by=["similarity"], ascending=False)
-        st.dataframe(words_df)
+
+words_df = pd.DataFrame(
+    st.session_state['words'],
+    columns=["word", "similarity"]
+).sort_values(by=["similarity"], ascending=False)
+st.dataframe(words_df)