Update app.py

app.py

@@ -18,12 +18,15 @@ st.write('Try to guess a secret word by semantic similarity')
 
 word = st.text_input("Input a word")
 
+used_words = [w for w, s in st.session_state['words']]
+
 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)
+    if word not in used_words:
+        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)