update handler

handler.py

@@ -71,7 +71,7 @@ class EndpointHandler():
         with torch.no_grad():
             for i, utt in enumerate(utterances):
                 if utt["speaker"] == params["speaker_2"] and (prev_num_words >= params["speaker_1_min_num_words"]):
-                    uptake_scores[utt["id"]] = self.get_uptake_score(textA=prev_text, textB=utt["text"])
+                    uptake_scores[str(utt["id"])] = self.get_uptake_score(textA=prev_text, textB=utt["text"])
                 prev_num_words = get_num_words(utt["text"])
                 prev_text = utt["text"]
 

test.py

@@ -6,8 +6,8 @@ my_handler = EndpointHandler(path=".")
 # 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" }
+   {"id": "1", "speaker": "Alice", "text": "How much is the fish?" },
+   {"id": "2", "speaker": "Bob", "text": "I have no idea, ask Alice" }
  ],
  "parameters": {
    "speaker_1_min_num_words": 5,