Spaces:

senatus-admin
/

senatus-dev

Running

dzenzzz commited on Mar 3

Commit

6644c19

1 Parent(s): 84fd3b1

minor change

Files changed (1) hide show

neural_searcher.py CHANGED Viewed

@@ -15,7 +15,7 @@ class NeuralSearcher:
     async def search(self, text: str):
-        dense_query = next(iter(self.dense_model.encode(text)))
         sparse_query = next(self.sparse_model.query_embed(text))
         late_query = next(self.late_interaction_model.query_embed(text))

     async def search(self, text: str):
+        dense_query = self.dense_model.encode(text).tolist()
         sparse_query = next(self.sparse_model.query_embed(text))
         late_query = next(self.late_interaction_model.query_embed(text))