Update app.py

app.py

@@ -57,6 +57,7 @@ if 'words_umap_df' not in st.session_state:
         "s": [],
         "l": [],
     })
+    st.session_state['words_umap_df'] = words_umap_df
     secret_embedding_3d = umap_3d.transform([secret_embedding])[0]
     words_umap_df.loc[len(words_umap_df)] = {
         "x": secret_embedding_3d[0],