GokulRajaR commited on
Commit
27e2ef7
·
verified ·
1 Parent(s): d6f2dc1

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +1 -1
server.py CHANGED
@@ -17,7 +17,7 @@ class EmbeddingAPI(ls.LitAPI):
17
  return request
18
 
19
  def predict(self, query):
20
- return self.model.encode(query)
21
 
22
  def encode_response(self, output):
23
  return output.tolist()
 
17
  return request
18
 
19
  def predict(self, query):
20
+ return self.model.encode_query(query)
21
 
22
  def encode_response(self, output):
23
  return output.tolist()