grantpitt commited on
Commit
e4a55aa
1 Parent(s): 4ddac9a

test redeployment

Browse files
Files changed (1) hide show
  1. handler.py +2 -3
handler.py CHANGED
@@ -17,9 +17,8 @@ class EndpointHandler:
17
  Return:
18
  A :obj:`list` | `dict`: will be serialized and returned
19
  """
20
-
21
- # pseudo
22
- token_inputs = self.tokenizer(data["inputs"], padding=True, return_tensors="pt")
23
  query_embed = self.model.get_text_features(**token_inputs)
24
  np_query_embed = query_embed.detach().cpu().numpy()[0].tolist()
25
  return np_query_embed
 
17
  Return:
18
  A :obj:`list` | `dict`: will be serialized and returned
19
  """
20
+ print("doesn this even get updated:", data)
21
+ token_inputs = self.tokenizer([data["inputs"]], padding=True, return_tensors="pt")
 
22
  query_embed = self.model.get_text_features(**token_inputs)
23
  np_query_embed = query_embed.detach().cpu().numpy()[0].tolist()
24
  return np_query_embed