SlowPacer commited on
Commit
f52fb63
1 Parent(s): 8a16697

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -35,7 +35,7 @@ class EndpointHandler():
35
  inputs = data.pop("inputs", data)
36
  parameters = data.pop("parameters", {})
37
 
38
- inputs = isinstance(inputs, list) or [inputs]
39
  raw_images = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
40
 
41
  processed_images = self.processor(images=raw_images, return_tensors="pt")
 
35
  inputs = data.pop("inputs", data)
36
  parameters = data.pop("parameters", {})
37
 
38
+ inputs = isinstance(inputs, list) and inputs or [inputs]
39
  raw_images = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
40
 
41
  processed_images = self.processor(images=raw_images, return_tensors="pt")