Spaces:

Klnimri
/

AI_CV_Matching

Running

Klnimri commited on Jan 13

Commit

bdb8ff0

verified ·

1 Parent(s): db3bd3d

app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ import docx2txt
 # Models
 # =========================================================
 EMBED_MODEL_NAME = os.getenv("EMBED_MODEL_NAME", "BAAI/bge-base-en-v1.5")
-RERANK_MODEL_NAME = os.getenv("RERANK_MODEL_NAME", "BAAI/bge-reranker-large")
 LLM_MODEL = os.getenv("LLM_MODEL", "meta-llama/Meta-Llama-3.1-8B-Instruct")

 # Models
 # =========================================================
 EMBED_MODEL_NAME = os.getenv("EMBED_MODEL_NAME", "BAAI/bge-base-en-v1.5")
+RERANK_MODEL_NAME = os.getenv("RERANK_MODEL_NAME", "BAAI/bge-reranker-base")
 LLM_MODEL = os.getenv("LLM_MODEL", "meta-llama/Meta-Llama-3.1-8B-Instruct")