Eyalyoli commited on
Commit
2535101
1 Parent(s): c26a43c

fix handler

Browse files
Files changed (1) hide show
  1. handler.py +4 -2
handler.py CHANGED
@@ -18,8 +18,10 @@ class EndpointHandler:
18
  A :obj:`list` | `dict`: will be serialized and returned
19
  """
20
  # get inputs
21
- texts = data.pop("texts", data)
22
- instruction = data.pop("instruction", "Represent this sentence:")
 
 
23
  # if isinstance(inputs, str):
24
  # inputs = [inputs]
25
 
 
18
  A :obj:`list` | `dict`: will be serialized and returned
19
  """
20
  # get inputs
21
+ inputs: dict = data.pop("inputs", data)
22
+ texts = inputs.pop("texts", None)
23
+ instruction = inputs.pop("instruction", None)
24
+
25
  # if isinstance(inputs, str):
26
  # inputs = [inputs]
27