Update apps/intro.py

apps/intro.py

@@ -8,8 +8,9 @@ import os
 import joblib
 
 def app():
+    word_to_embed_list = ["first"]
     with st.container():
-        question = st.text_input("Please enter your text here and we will embed it for you.",
+        word_to_embed= st.text_input("Please enter your text here and we will embed it for you.",
                                          value="Woman",)
         
         if st.button("Embed"):
@@ -19,8 +20,9 @@ def app():
                 REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
                 FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
                 umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
-    
-                examples_embeddings = model.encode(question)
+                
+                word_to_embed_list.append(word_to_embed)
+                examples_embeddings = model.encode(word_to_embed_list)
  
                 examples_umap = umap_model.transform(examples_embeddings)
 
@@ -30,7 +32,7 @@ def app():
                       fig = px.scatter_3d(
                           examples_umap , x=0, y=1, z=2,
                            # color='labels',
-                          opacity = .5,    hover_data=[question])
+                          opacity = .5,    hover_data=[word_to_embed_list])
                       fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
                       fig.update_traces(marker_size=4)
                       st.plotly_chart(fig)