Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def transform_query(query: str) -> str:
|
|
55 |
return f'Represent this sentence for searching relevant passages: {query}'
|
56 |
|
57 |
def query_hybrid_search(query: str, client: QdrantClient, collection_name: str, dense_model: OptimumEncoder, sparse_model: SparseTextEmbedding):
|
58 |
-
dense_embeddings = dense_model([transform_query(query)], 1,
|
59 |
sparse_embeddings = list(sparse_model.query_embed(query))[0]
|
60 |
|
61 |
return client.query_points(
|
@@ -301,7 +301,7 @@ def chunk_documents(texts: List[str], metadatas: List[dict], dense_model: Optimu
|
|
301 |
print(f'CHUNKS : {documents}')
|
302 |
|
303 |
start_dense = time.time()
|
304 |
-
dense_embeddings = dense_model(documents, 32,
|
305 |
end_dense = time.time()
|
306 |
final_dense = end_dense - start_dense
|
307 |
print(f'DENSE TIME: {final_dense}')
|
|
|
55 |
return f'Represent this sentence for searching relevant passages: {query}'
|
56 |
|
57 |
def query_hybrid_search(query: str, client: QdrantClient, collection_name: str, dense_model: OptimumEncoder, sparse_model: SparseTextEmbedding):
|
58 |
+
dense_embeddings = dense_model([transform_query(query)], 1, convert_to_numpy=True)[0]
|
59 |
sparse_embeddings = list(sparse_model.query_embed(query))[0]
|
60 |
|
61 |
return client.query_points(
|
|
|
301 |
print(f'CHUNKS : {documents}')
|
302 |
|
303 |
start_dense = time.time()
|
304 |
+
dense_embeddings = dense_model(documents, 32, convert_to_numpy=True)
|
305 |
end_dense = time.time()
|
306 |
final_dense = end_dense - start_dense
|
307 |
print(f'DENSE TIME: {final_dense}')
|