test redeployment
Browse files- 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 |
-
|
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
|