Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,8 +62,8 @@ async def lifespan(app: FastAPI):
|
|
| 62 |
|
| 63 |
# Load models into the shared dictionary
|
| 64 |
ml_models["embedder"] = HuggingFaceEmbeddings(
|
| 65 |
-
|
| 66 |
-
model_name="intfloat/e5-large-v2", #lil faster
|
| 67 |
encode_kwargs={
|
| 68 |
"batch_size": 64,
|
| 69 |
# "normalize_embeddings": True
|
|
@@ -167,7 +167,7 @@ async def run_hackrx(req: RunRequest):
|
|
| 167 |
)
|
| 168 |
|
| 169 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 170 |
-
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k":
|
| 171 |
|
| 172 |
|
| 173 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|
|
|
|
| 62 |
|
| 63 |
# Load models into the shared dictionary
|
| 64 |
ml_models["embedder"] = HuggingFaceEmbeddings(
|
| 65 |
+
model_name="BAAI/bge-base-en-v1.5", #better but lil slower
|
| 66 |
+
# model_name="intfloat/e5-large-v2", #lil faster
|
| 67 |
encode_kwargs={
|
| 68 |
"batch_size": 64,
|
| 69 |
# "normalize_embeddings": True
|
|
|
|
| 167 |
)
|
| 168 |
|
| 169 |
# dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8})
|
| 170 |
+
dense_retriever = vectorstore.as_retriever(search_type="mmr",search_kwargs={"k": 8 ,"lambda_mult": 0.5})
|
| 171 |
|
| 172 |
|
| 173 |
# Create retrievers using the pre-loaded models from our ml_models dictionary
|