stanford-nlpxed
/

uptake-model

Inference Endpoints

Model card Files Files and versions Community

ddemszky commited on Jan 20, 2023

Commit

abd286b

•

1 Parent(s): ba5a35c

update handler

Files changed (2) hide show

handler.py +1 -1
test.py +1 -1

handler.py CHANGED Viewed

@@ -56,7 +56,7 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
-        utterances = data.pop("utterances", data)
         params = data.pop("parameters", None)
         print("EXAMPLES")

             A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
+        utterances = data.pop("inputs", data)
         params = data.pop("parameters", None)
         print("EXAMPLES")

test.py CHANGED Viewed

@@ -5,7 +5,7 @@ my_handler = EndpointHandler(path=".")
 # prepare sample payload
 example = {
- "utterances": [
    {"id": 1, "speaker": "Alice", "text": "How much is the fish?" },
    {"id": 2, "speaker": "Bob", "text": "I have no idea, ask Alice" }
  ],

 # prepare sample payload
 example = {
+ "inputs": [
    {"id": 1, "speaker": "Alice", "text": "How much is the fish?" },
    {"id": 2, "speaker": "Bob", "text": "I have no idea, ask Alice" }
  ],