LDanielBlueway commited on
Commit
550a31d
·
verified ·
1 Parent(s): 2c8aa2f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -0
handler.py CHANGED
@@ -3,6 +3,7 @@ from PIL import Image
3
  from io import BytesIO
4
  from transformers import AutoProcessor, OmDetTurboForObjectDetection
5
  import base64
 
6
 
7
  class EndpointHandler():
8
  def __init__(self, path=""):
@@ -32,6 +33,8 @@ class EndpointHandler():
32
  score_threshold=0.3,
33
  nms_threshold=0.3,
34
  )[0]
 
 
35
  # Convert tensors to lists
36
  serializable_results = {
37
  'boxes': results['boxes'].tolist(),
 
3
  from io import BytesIO
4
  from transformers import AutoProcessor, OmDetTurboForObjectDetection
5
  import base64
6
+ import logging
7
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
 
33
  score_threshold=0.3,
34
  nms_threshold=0.3,
35
  )[0]
36
+ print(results)
37
+ logging.error(results)
38
  # Convert tensors to lists
39
  serializable_results = {
40
  'boxes': results['boxes'].tolist(),