fashxp commited on
Commit
e7656af
1 Parent(s): ae0228d

update params

Browse files
Files changed (1) hide show
  1. handler.py +7 -3
handler.py CHANGED
@@ -17,11 +17,15 @@ class EndpointHandler():
17
  Return:
18
  A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
19
  """
20
- inputs = data.pop("inputs", data)
 
 
 
21
  # decode base64 image to PIL
22
- image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
23
 
24
- parameters = inputs['parameters']
 
25
  candidate_labels = parameters['candidate_labels']
26
 
27
  # run prediction one image wit provided candiates
 
17
  Return:
18
  A :obj:`list`:. The list contains items that are dicts should be liked {"label": "XXX", "score": 0.82}
19
  """
20
+ #inputs = data.pop("inputs", data)
21
+ #image_data = inputs['image']
22
+
23
+ image_data = data.pop("image", data)
24
  # decode base64 image to PIL
25
+ image = Image.open(BytesIO(base64.b64decode(image_data)))
26
 
27
+ parameters = data.pop("parameters", data)
28
+ #parameters = inputs['parameters']
29
  candidate_labels = parameters['candidate_labels']
30
 
31
  # run prediction one image wit provided candiates