ryanrwatkins commited on
Commit
0569eb0
1 Parent(s): 78b8854

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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,