Allob commited on
Commit
ed2d3e4
·
1 Parent(s): 91b81a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ umap_3d = prepare_umap()
40
 
41
 
42
  secret_word =choose_secret_word()
43
- secret_embedding = model.encode(secret_word)
44
 
45
  print("Secret word ", secret_word)
46
 
@@ -79,7 +79,7 @@ used_words = [w for w, s in st.session_state['words']]
79
 
80
  if st.button("Guess") or word:
81
  if word not in used_words:
82
- word_embedding = model.encode(word)
83
  similarity = util.pytorch_cos_sim(
84
  secret_embedding,
85
  word_embedding
 
40
 
41
 
42
  secret_word =choose_secret_word()
43
+ secret_embedding = model.encode(secret_word.lower().strip())
44
 
45
  print("Secret word ", secret_word)
46
 
 
79
 
80
  if st.button("Guess") or word:
81
  if word not in used_words:
82
+ word_embedding = model.encode(word.lower().strip())
83
  similarity = util.pytorch_cos_sim(
84
  secret_embedding,
85
  word_embedding