luciagomez commited on
Commit
7a93c4f
·
verified ·
1 Parent(s): aee999b

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +3 -3
retriever.py CHANGED
@@ -18,7 +18,7 @@ DATASET_REPO = "luciagomez/MrPhil_vector"
18
  # -------------------------------------------------------------------
19
  parquet_path = hf_hub_download(
20
  repo_id=DATASET_REPO,
21
- filename="foundations.parquet",
22
  repo_type="dataset",
23
  token=HF_TOKEN_read,
24
  cache_dir="/tmp/huggingface"
@@ -26,7 +26,7 @@ parquet_path = hf_hub_download(
26
 
27
  faiss_path = hf_hub_download(
28
  repo_id=DATASET_REPO,
29
- filename="faiss.index",
30
  repo_type="dataset",
31
  token=HF_TOKEN_read,
32
  cache_dir="/tmp/huggingface"
@@ -34,7 +34,7 @@ faiss_path = hf_hub_download(
34
 
35
  meta_path = hf_hub_download(
36
  repo_id=DATASET_REPO,
37
- filename="meta.json",
38
  repo_type="dataset",
39
  token=HF_TOKEN_read,
40
  cache_dir="/tmp/huggingface"
 
18
  # -------------------------------------------------------------------
19
  parquet_path = hf_hub_download(
20
  repo_id=DATASET_REPO,
21
+ filename="bgem3/foundations.parquet",
22
  repo_type="dataset",
23
  token=HF_TOKEN_read,
24
  cache_dir="/tmp/huggingface"
 
26
 
27
  faiss_path = hf_hub_download(
28
  repo_id=DATASET_REPO,
29
+ filename="bgem3/faiss.index",
30
  repo_type="dataset",
31
  token=HF_TOKEN_read,
32
  cache_dir="/tmp/huggingface"
 
34
 
35
  meta_path = hf_hub_download(
36
  repo_id=DATASET_REPO,
37
+ filename="bgem3/meta.json",
38
  repo_type="dataset",
39
  token=HF_TOKEN_read,
40
  cache_dir="/tmp/huggingface"