Update apps/intro.py

apps/intro.py

@@ -18,13 +18,19 @@ def app():
                 model = SentenceTransformer(model_name)
                 REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
                 FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
-                st.write("load umap")
-                model_umap = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
-                st.write("embed umap")
- 
-                
+                umap_model= joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
+    
                 examples_embeddings = model.encode(question)
-                st.write("umap")
-                examples_umap = model_umap.transform(examples_embeddings)
+ 
+                examples_umap = umap_model.transform(examples_embeddings)
 
-                st.write(len(examples_umap))
+                #st.write(len(examples_umap))
+                
+                with st.spinner("👑 create visualisation"):  
+                      fig = px.scatter_3d(
+                          examples_umap , x=0, y=1, z=2,
+                           # color='labels',
+                          opacity = .5,    hover_data=[question])
+                      fig.update_scenes(xaxis_visible=False, yaxis_visible=False,zaxis_visible=False )
+                      fig.update_traces(marker_size=4)
+                      st.plotly_chart(fig)