devve1 commited on
Commit
6b9d2ce
1 Parent(s): 7e270b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, convert_to_numpy=True)[0]
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, convert_to_numpy=True)
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, 0, 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, 0, convert_to_numpy=True)
305
  end_dense = time.time()
306
  final_dense = end_dense - start_dense
307
  print(f'DENSE TIME: {final_dense}')