devve1 commited on
Commit
1261a90
1 Parent(s): 0208e3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def transform_query(query: str) -> str:
35
  return f'Represent this sentence for searching relevant passages: {query}'
36
 
37
  def query_hybrid_search(col: Collection, query: str, dense_model: OptimumEncoder, sparse_model: Bm42):
38
- query_dense_embeddings = dense_model(transform_query(query), 1)
39
  query_sparse_embeddings = list(sparse_model.query_embed(query))
40
 
41
  query_sparse = query_sparse_embeddings[0]
 
35
  return f'Represent this sentence for searching relevant passages: {query}'
36
 
37
  def query_hybrid_search(col: Collection, query: str, dense_model: OptimumEncoder, sparse_model: Bm42):
38
+ query_dense_embeddings = dense_model([transform_query(query)], 1)
39
  query_sparse_embeddings = list(sparse_model.query_embed(query))
40
 
41
  query_sparse = query_sparse_embeddings[0]