Update apps/intro.py

apps/intro.py

@@ -1,6 +1,7 @@
 import plotly.express as px
 import streamlit as st
 from sentence_transformers import SentenceTransformer
+from huggingface_hub import hf_hub_url, cached_download
 import umap.umap_ as umap
 import pandas as pd
 import os
@@ -15,8 +16,10 @@ def app():
             with st.spinner("👑 load language model (sentence transformer)"):
                 model_name = 'sentence-transformers/all-MiniLM-L6-v2'
                 model = SentenceTransformer(model_name)
-                umap_name = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
-                umap_model = joblib.load(umap_name)
+                REPO_ID = "peter2000/umap_embed_3d_all-MiniLM-L6-v2"
+                FILENAME = "umap_embed_3d_all-MiniLM-L6-v2.sav"
+                model_umap = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
+                
                 docs_umap = umap_model.transform(docs_embeddings)
                 
                 examples_embeddings = model.encode(question)