update handler

handler.py

@@ -19,6 +19,7 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
+        text = data.pop("text", data)
         tokens = self.tokenizer(text, return_tensors='pt')
         output = self.model(**tokens)
         vec = torch.max(
@@ -26,7 +27,6 @@ class EndpointHandler():
                 1 + torch.relu(output.logits)
             ) * tokens.attention_mask.unsqueeze(-1),
             dim=1)[0].squeeze()
-        instruction = data.pop("instruction", data)
         cols = vec.nonzero().squeeze().cpu().tolist()
         # extract the non-zero values
         weights = vec[cols].cpu().tolist()