Allob commited on
Commit
418bcdd
·
1 Parent(s): ee5ab0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ if st.button("Guess") or word:
57
  if word not in used_words:
58
  word_embedding = [models[name].encode(word.lower().strip()) for name in model_names]
59
  similarities = [util.pytorch_cos_sim(secret_embedding[i], word_embedding[i]).cpu().numpy()[0][0] for i, name in enumerate(model_names)]
60
- st.session_state['words'].append([str(word)] + similarities))
61
 
62
 
63
 
 
57
  if word not in used_words:
58
  word_embedding = [models[name].encode(word.lower().strip()) for name in model_names]
59
  similarities = [util.pytorch_cos_sim(secret_embedding[i], word_embedding[i]).cpu().numpy()[0][0] for i, name in enumerate(model_names)]
60
+ st.session_state['words'].append([str(word)] + similarities)
61
 
62
 
63