Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|