Update apps/intro.py

apps/intro.py

@@ -8,7 +8,7 @@ import os
 import joblib
 
 def app():
-    word_to_embed_list = ["first"]
+    word_to_embed_list =  st.session_state['embed_list']
     with st.container():
         word_to_embed= st.text_input("Please enter your text here and we will embed it for you.",
                                          value="Woman",)
@@ -22,6 +22,8 @@ def app():
                 umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
                 
                 word_to_embed_list.append(word_to_embed)
+                st.session_state['embed_list'] = word_to_embed_list
+                
                 examples_embeddings = model.encode(word_to_embed_list)
  
                 examples_umap = umap_model.transform(examples_embeddings)
@@ -30,9 +32,9 @@ def app():
                 
                 with st.spinner("👑 create visualisation"):  
                       fig = px.scatter_3d(
-                          examples_umap , x=0, y=1, z=2,
+                          examples_umap[1:] , x=0, y=1, z=2,
                            # color='labels',
-                          opacity = .5,    hover_data=[word_to_embed_list])
+                          opacity = .7,    hover_data=[word_to_embed_list[1:]])
                       fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
                       fig.update_traces(marker_size=4)
                       st.plotly_chart(fig)