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
|