LDanielBlueway commited on
Commit
841d06b
·
verified ·
1 Parent(s): 550a31d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -39,7 +39,7 @@ class EndpointHandler():
39
  serializable_results = {
40
  'boxes': results['boxes'].tolist(),
41
  'scores': results['scores'].tolist(),
42
- 'text_labels': results['text_labels'] # Already serializable
43
  }
44
  return serializable_results
45
  #prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])
 
39
  serializable_results = {
40
  'boxes': results['boxes'].tolist(),
41
  'scores': results['scores'].tolist(),
42
+ 'candidates': results['classes'] # Already serializable
43
  }
44
  return serializable_results
45
  #prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])