Initial codes commit
Browse files- 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
|