peter2000 commited on
Commit
4258e95
1 Parent(s): c06e027

Update apps/intro.py

Browse files
Files changed (1) hide show
  1. apps/intro.py +4 -2
apps/intro.py CHANGED
@@ -18,11 +18,13 @@ def app():
18
  model = SentenceTransformer(model_name)
19
  REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
20
  FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
 
21
  model_umap = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
22
-
23
- docs_umap = umap_model.transform(docs_embeddings)
24
 
25
  examples_embeddings = model.encode(question)
 
26
  examples_umap = umap_model.transform(examples_embeddings)
27
 
28
  st.write(examples_umap.shape)
 
18
  model = SentenceTransformer(model_name)
19
  REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
20
  FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
21
+ st.write("load umap")
22
  model_umap = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
23
+ st.write("embed umap")
24
+
25
 
26
  examples_embeddings = model.encode(question)
27
+ st.write("umap")
28
  examples_umap = umap_model.transform(examples_embeddings)
29
 
30
  st.write(examples_umap.shape)