rishadaz commited on
Commit
276d404
Β·
verified Β·
1 Parent(s): 877724a

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +8 -2
src/streamlit_app.py CHANGED
@@ -44,10 +44,16 @@ from huggingface_hub import hf_hub_download, snapshot_download
44
 
45
  VECTOR_STORE_DIR = ROOT / "embeddings" / "semantic_vector_store"
46
 
47
- VECTOR_STORE_DIR = Path("/data/embeddings/semantic_vector_store")
48
-
49
  @st.cache_resource
50
  def load_vector_store_cached():
 
 
 
 
 
 
 
 
51
  return load_vector_store(VECTOR_STORE_DIR)
52
 
53
  # ─── Custom CSS ───────────────────────────────────────────────────────────────
 
44
 
45
  VECTOR_STORE_DIR = ROOT / "embeddings" / "semantic_vector_store"
46
 
 
 
47
  @st.cache_resource
48
  def load_vector_store_cached():
49
+ VECTOR_STORE_DIR.mkdir(parents=True, exist_ok=True)
50
+
51
+ snapshot_download(
52
+ repo_id="rishadaz/amazon_retriever-storage",
53
+ repo_type="dataset",
54
+ local_dir=str(VECTOR_STORE_DIR),
55
+ )
56
+
57
  return load_vector_store(VECTOR_STORE_DIR)
58
 
59
  # ─── Custom CSS ───────────────────────────────────────────────────────────────