Loren commited on
Commit
637d2cd
·
verified ·
1 Parent(s): 008ba0c

Update app/database.py

Browse files
Files changed (1) hide show
  1. app/database.py +1 -1
app/database.py CHANGED
@@ -19,7 +19,7 @@ REPO_ID = "Loren/articles_database"
19
  FAISS_REPO_ID = "Loren/articles_faiss"
20
  FAISS_INDEX_FILE = "faiss_index.bin"
21
  MODEL_NAME = "intfloat/multilingual-e5-small"
22
- CROSS_ENCODER_NAME = "jinaai/jina-reranker-v2-base-multilingual"
23
 
24
  cache_dir = "/tmp"
25
  os.makedirs(cache_dir, exist_ok=True)
 
19
  FAISS_REPO_ID = "Loren/articles_faiss"
20
  FAISS_INDEX_FILE = "faiss_index.bin"
21
  MODEL_NAME = "intfloat/multilingual-e5-small"
22
+ CROSS_ENCODER_NAME = "cross-encoder/ms-marco-MiniLM-L12-v2"
23
 
24
  cache_dir = "/tmp"
25
  os.makedirs(cache_dir, exist_ok=True)