zaddyzaddy commited on
Commit
c577cbf
·
verified ·
1 Parent(s): b20b21e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. handler.py +2 -0
handler.py CHANGED
@@ -26,6 +26,8 @@ class EndpointHandler:
26
  Return:
27
  A :obj:`list` | `dict`: will be serialized and returned
28
  """
 
 
29
  input_text = data.get("input_text", "")
30
  lex_diversity = data.get("lex_diversity", 80)
31
  order_diversity = data.get("order_diversity", 20)
 
26
  Return:
27
  A :obj:`list` | `dict`: will be serialized and returned
28
  """
29
+
30
+ data = data.pop("inputs", data)
31
  input_text = data.get("input_text", "")
32
  lex_diversity = data.get("lex_diversity", 80)
33
  order_diversity = data.get("order_diversity", 20)