Hitan2004 commited on
Commit
60f2a58
·
verified ·
1 Parent(s): edf680f

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +2 -2
config.py CHANGED
@@ -17,8 +17,8 @@ FAISS_INDEX_PATH = os.path.join(_BASE, "faiss.index")
17
  BM25_PATH = os.path.join(_BASE, "bm25.pkl")
18
  CHUNKS_PATH = os.path.join(_BASE, "chunks.pkl")
19
  SOURCES_PATH = os.path.join(_BASE, "sources.pkl")
20
- EMBEDDER_NAME = "all-MiniLM-L6-v2" # ← correct, HF model name
21
- RERANKER_MODEL = os.path.join(_BASE, "models/reranker")
22
  CHUNK_SIZE = 500
23
  CHUNK_OVERLAP = 50
24
  TOP_K = 5
 
17
  BM25_PATH = os.path.join(_BASE, "bm25.pkl")
18
  CHUNKS_PATH = os.path.join(_BASE, "chunks.pkl")
19
  SOURCES_PATH = os.path.join(_BASE, "sources.pkl")
20
+ EMBEDDER_NAME = "all-MiniLM-L6-v2"
21
+ RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2"
22
  CHUNK_SIZE = 500
23
  CHUNK_OVERLAP = 50
24
  TOP_K = 5