takanezawa commited on
Commit
4f6d751
·
1 Parent(s): 8b84237

chore: Refactor handler.py to return a list of predictions

Browse files
Files changed (1) hide show
  1. handler.py +5 -3
handler.py CHANGED
@@ -39,6 +39,8 @@ class EndpointHandler():
39
  # else:
40
  # prediction = self.pipeline(inputs)
41
  # postprocess the prediction
42
- return {
43
- "sparse_embeddings": "12345"
44
- }
 
 
 
39
  # else:
40
  # prediction = self.pipeline(inputs)
41
  # postprocess the prediction
42
+ return [
43
+ [
44
+ {"label": 0, "score": 0.5}
45
+ ]
46
+ ]