Upload handler.py
Browse files- 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 |
-
|
89 |
-
|
90 |
-
|
|
|
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)
|