Update app.py

app.py

@@ -13,6 +13,7 @@ with st.spinner("👑 load language model (sentence transformer)"):
 
 with st.spinner("👑 load sdg data"):
   df_osdg = pd.read_csv('https://zenodo.org/record/5550238/files/osdg-community-dataset-v21-09-30.csv',sep='\t')
+  df_osdg  = df_osdg[:1000]
   
   _lab_dict = {0: 'no_cat',
               1:'SDG 1 - No poverty',
@@ -38,7 +39,7 @@ with st.spinner("👑 load sdg data"):
   docs = list(df_osdg['text'])
   docs_embeddings = model.encode(docs)
 
-with st.spinner("👑 create visualisation"):
+with st.spinner("👑 prepare visualisation"):
   n_neighbors = 15
   n_components = 3
   random_state =42
@@ -49,7 +50,8 @@ with st.spinner("👑 create visualisation"):
                           .fit(docs_embeddings))
                           
   docs_umap = umap_model.transform(docs_embeddings)
-  
+
+with st.spinner("👑 create visualisation"):  
   fig = px.scatter_3d(
       docs_umap, x=0, y=1, z=2,
       color=labels,