Update app.py
Browse files
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]
|