m97j commited on
Commit
9f3b43b
ยท
1 Parent(s): 95c3e86

Initial codes commit

Browse files
Files changed (1) hide show
  1. db/initializer.py +2 -2
db/initializer.py CHANGED
@@ -10,8 +10,8 @@ _vector_ids = None
10
 
11
  def _load_index_in_memory():
12
  """HF Hub์—์„œ ์ธ๋ฑ์Šค/ID ๋งคํ•‘์„ ๋ฐ›์•„ ๋ฉ”๋ชจ๋ฆฌ์— ๋กœ๋“œ"""
13
- index_path = hf_hub_download(repo_id=HF_REPO_ID, filename=HF_INDEX_FILE)
14
- ids_path = hf_hub_download(repo_id=HF_REPO_ID, filename=HF_IDS_FILE)
15
  index = faiss.read_index(index_path)
16
  set_index(index)
17
  global _vector_ids
 
10
 
11
  def _load_index_in_memory():
12
  """HF Hub์—์„œ ์ธ๋ฑ์Šค/ID ๋งคํ•‘์„ ๋ฐ›์•„ ๋ฉ”๋ชจ๋ฆฌ์— ๋กœ๋“œ"""
13
+ index_path = hf_hub_download(repo_id=HF_REPO_ID, filename=HF_INDEX_FILE, repo_type="dataset")
14
+ ids_path = hf_hub_download(repo_id=HF_REPO_ID, filename=HF_IDS_FILE, repo_type="dataset")
15
  index = faiss.read_index(index_path)
16
  set_index(index)
17
  global _vector_ids