krilecy commited on
Commit
bbeb214
1 Parent(s): a2cbf75

Upload handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -3
handler.py CHANGED
@@ -85,9 +85,10 @@ class EndpointHandler():
85
  Return:
86
  A :obj:`list` | `dict`: will be serialized and returned
87
  """
88
- id = data.pop("id", data)
89
- text = data.pop("text", data)
90
- request_type = data.pop("type", data)
 
91
 
92
 
93
  embeddings, attn_vec = self.embed(text, request_type)
 
85
  Return:
86
  A :obj:`list` | `dict`: will be serialized and returned
87
  """
88
+ inputs = data.pop("inputs", data)
89
+ id = inputs.pop("id", inputs)
90
+ text = inputs.pop("text", inputs)
91
+ request_type = inputs.pop("type", inputs)
92
 
93
 
94
  embeddings, attn_vec = self.embed(text, request_type)