sergeipetrov HF staff commited on
Commit
696a8e1
1 Parent(s): 42762fe

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -3
handler.py CHANGED
@@ -33,10 +33,8 @@ class EndpointHandler():
33
  """
34
  inputs = data.pop("inputs", data)
35
  parameters = data.pop("parameters", {})
36
-
37
- raw_images = [Image.open(BytesIO(_img)) for _img in inputs]
38
 
39
- processed_image = self.processor(images=raw_images, return_tensors="pt")
40
  processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
41
  processed_image = {**processed_image, **parameters}
42
 
 
33
  """
34
  inputs = data.pop("inputs", data)
35
  parameters = data.pop("parameters", {})
 
 
36
 
37
+ processed_image = self.processor(images=inputs, return_tensors="pt")
38
  processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
39
  processed_image = {**processed_image, **parameters}
40