dzenzzz commited on
Commit
2acaa2a
·
1 Parent(s): c6c4707

minor change

Browse files
Files changed (1) hide show
  1. neural_searcher.py +1 -1
neural_searcher.py CHANGED
@@ -15,7 +15,7 @@ class NeuralSearcher:
15
 
16
  async def search(self, text: str):
17
 
18
- dense_query = next(self.dense_model.encode(text).tolist())
19
  sparse_query = next(self.sparse_model.query_embed(text))
20
  late_query = next(self.late_interaction_embedding_model.query_embed(text))
21
 
 
15
 
16
  async def search(self, text: str):
17
 
18
+ dense_query = next(iter(self.dense_model.encode(text)))
19
  sparse_query = next(self.sparse_model.query_embed(text))
20
  late_query = next(self.late_interaction_embedding_model.query_embed(text))
21