Spaces:
Building
Building
Update app.py
Browse files
app.py
CHANGED
@@ -383,8 +383,7 @@ def create_compression_retriever(embeddings, base_retriever, chunk_size=500, k=1
|
|
383 |
print("compression retriever done")
|
384 |
return compression_retriever
|
385 |
|
386 |
-
def CohereRerank_retriever(
|
387 |
-
print("cohere rerank started")
|
388 |
base_retriever,
|
389 |
cohere_api_key,cohere_model="rerank-multilingual-v2.0", top_n=8
|
390 |
):
|
@@ -395,7 +394,7 @@ def CohereRerank_retriever(
|
|
395 |
cohere_model: The Cohere model can be either 'rerank-english-v2.0' or 'rerank-multilingual-v2.0', with the latter being the default.
|
396 |
top_n: top n results returned by Cohere rerank, default = 8.
|
397 |
"""
|
398 |
-
|
399 |
compressor = CohereRerank(
|
400 |
cohere_api_key=cohere_api_key,
|
401 |
model=cohere_model,
|
|
|
383 |
print("compression retriever done")
|
384 |
return compression_retriever
|
385 |
|
386 |
+
def CohereRerank_retriever(
|
|
|
387 |
base_retriever,
|
388 |
cohere_api_key,cohere_model="rerank-multilingual-v2.0", top_n=8
|
389 |
):
|
|
|
394 |
cohere_model: The Cohere model can be either 'rerank-english-v2.0' or 'rerank-multilingual-v2.0', with the latter being the default.
|
395 |
top_n: top n results returned by Cohere rerank, default = 8.
|
396 |
"""
|
397 |
+
print("cohere rerank started")
|
398 |
compressor = CohereRerank(
|
399 |
cohere_api_key=cohere_api_key,
|
400 |
model=cohere_model,
|