asoderznik commited on
Commit
58dfe8e
1 Parent(s): 9a7b414

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -51,4 +51,4 @@ class EndpointHandler():
51
  #img_str = base64.b64encode(buffered.getvalue())
52
 
53
  # postprocess the prediction
54
- return {"image": "img_str.decode()"}
 
51
  #img_str = base64.b64encode(buffered.getvalue())
52
 
53
  # postprocess the prediction
54
+ return {"image": inputs['image']}